Update app.py

app.py

@@ -6,9 +6,9 @@ import torch
 # Initialize FastAPI app
 app = FastAPI()
 
-# Load pre-trained DistilGPT-2 model and tokenizer
+# Load pre-trained DistilGPT-2 model and tokenizer, using from_tf=True to load TensorFlow weights
 model_name = "distilgpt2"  # Smaller GPT-2 model
-model = AutoModelForCausalLM.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name, from_tf=True)  # Use from_tf=True
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 
 # Pydantic model for request body