Upload app.py

app.py

@@ -8,9 +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():
-    # 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")
+    # 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 = torch.load(checkpoint_path, map_location=device)
     config = checkpoint['config']
@@ -68,20 +68,20 @@ def generate_text(prompt, max_length=100, num_samples=1, temperature=0.8):
 iface = gr.Interface(
     fn=generate_text,
     inputs=[
-        gr.Textbox(label="Prompt", value="We are accounted poor citizens, the"),
+        gr.Textbox(label="Prompt", value="Good night, good night! Parting is such sweet sorrow"),
         gr.Slider(minimum=10, maximum=200, value=100, step=1, label="Max Length"),
         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 a prompt to generate Shakespeare-style text continuation",
+    description="Enter text for Shakespear way of text continuation",
     examples=[
-        ["O Romeo, Romeo, wherefore art thou", 100, 1],
-        ["To be, or not to be, that is", 60, 2],
-        ["Friends, Romans, countrymen, lend me", 50, 3],
-        ["All the world's a stage, and all the", 100, 1],
-        ["Now is the winter of our discontent", 100, 1],
-        ["If music be the food of love,", 100, 1],
+        ["To be, or not to be: that is the question.", 100, 1],
+        ["Love all, trust a few, do wrong to none.", 60, 2],
+        ["It's not enough to speak, but to speak true", 50, 3],
+        ["There are more things in heaven and earth, Horatio, than are dreamt of in your philosophy.", 100, 1],
+        ["If you can look into the seeds of time, and say which grain will grow and which will not, speak then to me", 100, 1],
+        ["Love sought is good, but given unsought is better.", 100, 1],
     ]
 )