Update app.py

app.py

@@ -10,7 +10,7 @@ def get_model():
     model = GPT(GPTConfig())
     # Load from the Hugging Face Hub instead of local file
     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.load_state_dict(torch.hub.load_state_dict_from_url(f'https://huggingface.co/{model_path}/resolve/main/model.pth', map_location='cpu'))
     model.eval()
     return model