Update app.py

app.py

@@ -10,7 +10,7 @@ device = 'cuda' if torch.cuda.is_available() else 'cpu'
 # Load the model from Hugging Face Hub
 def load_model_from_huggingface():
     # Replace with your Hugging Face model ID (username/model-name)
-    model_id = "Rajendro/StyleDecoderTransformer"
+    model_id = "EzhirkoArulmozhi/DecoderTransformerModel"
     checkpoint_path = hf_hub_download(repo_id=model_id, filename="gpt_checkpoint.pth")
     
     checkpoint = torch.load(checkpoint_path, map_location=device)