Update app.py

app.py

@@ -9,8 +9,8 @@ from transformer import GPT, GPTConfig  # Import your model class
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 def load_model_from_hf():
     # Replace with your Hugging Face model ID (username/model-name)
-    model_id = "satyanayak/transformer-basic"
-    checkpoint_path = hf_hub_download(repo_id=model_id, filename="trained_model.pt")
+    model_id = "sudhakar272/transformer-basic"
+    checkpoint_path = hf_hub_download(repo_id=model_id, filename="transformer_model.pt")
     
     checkpoint = torch.load(checkpoint_path, map_location=device)
     config = checkpoint['config']