changed app.py

app.py

@@ -9,12 +9,12 @@ def get_model():
     """Load the trained GPT model."""
     model = GPT(GPTConfig())
     # Load from the Hugging Face Hub instead of local file
-    model_path = 'YOUR_USERNAME/YOUR_MODEL_REPO/final_best_model.pth'
-    model.load_state_dict(torch.hub.load_state_dict_from_url(f'https://huggingface.co/{model_path}/resolve/main/final_best_model.pth', map_location='cpu')['model_state_dict'])
+    model_path = 'mathminakshi/custom_gpt2'
+    model.load_state_dict(torch.hub.load_state_dict_from_url(f'https://huggingface.co/{model_path}/resolve/main/best_model.pth', map_location='cpu')['model_state_dict'])
     model.eval()
     return model
 
-def generate_text(prompt, max_tokens=500, temperature=0.8, top_k=40):
+def generate_text(prompt, max_tokens=500, temperature=0.3, top_k=40):
     """Generate text based on the prompt."""
     # Encode the prompt
     enc = tiktoken.get_encoding('gpt2')