Update app.py

app.py

@@ -9,7 +9,7 @@ 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 = "sudhakar272/transformer-basic"
+    model_id = "sudhakar272/transformer_model"
     checkpoint_path = hf_hub_download(repo_id=model_id, filename="transformer_model.pt")
     
     checkpoint = torch.load(checkpoint_path, map_location=device)