Upload app.py

app.py

@@ -8,11 +8,9 @@ from transformer import GPT, GPTConfig  # Import your model class
 # Load the model from Hugging Face Hub
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 def load_model_from_hf():
-    # Hugging Face model ID (username/model-name)
-    model_id = "sudhakar272/shakespheretextgenerator"
-    checkpoint_path = hf_hub_download(repo_id=model_id, filename="transformer_model.pt")
+    checkpoint_path ="./transformer_model.pt"
     
-    checkpoint = torch.load(checkpoint_path, map_location=device)
+    checkpoint = torch.load(checkpoint_path)
     config = checkpoint['config']
     model = GPT(config)
     model.load_state_dict(checkpoint['model_state_dict'])
@@ -73,8 +71,8 @@ iface = gr.Interface(
         gr.Slider(minimum=1, maximum=5, value=1, step=1, label="Number of Samples"),
     ],
     outputs=gr.Textbox(label="Generated Text"),
-    title="Shakespeare-style Text Generator",
-    description="Enter text for Shakespear way of text continuation",
+    title="Shakesphere Text Generator",
+    description="Enter text for Shakesphere way of text and continue the same",
     examples=[
         ["To be, or not to be: that is the question.", 100, 1],
         ["Love all, trust a few, do wrong to none.", 60, 2],