Upload 2 files

app.py

@@ -35,11 +35,10 @@ interface = gr.Interface(
     fn=gradio_interface,
     inputs=[
         gr.Textbox(label="Prompt", value="Once upon a time"),
-        gr.Slider(50, 300, step=1, value=75, label="Max Length"),
-        gr.Slider(0.1, 1.5, step=0.1, value=1.0, label="Temperature")
+        gr.Slider(50, 100, step=1, value=75, label="Max Length"),
     ],
     outputs="text",
-    title="GPT Text Generation",
+    title="Odeyssey Rhyme Generator",
     description="Enter a prompt to generate text."
 )
 

gpt_dev.py

@@ -425,7 +425,7 @@ a = a / torch.sum(a, 1, keepdim = True)
 b = torch.randint(0, 10, (3, 2)).float()
 c = a @ b
 
-def generate_text(model, start_prompt, max_length=300, temperature=1.0):
+def generate_text(model, start_prompt, max_length=100, temperature=1.0):
     input_ids = torch.tensor(encode(start_prompt), dtype=torch.long).unsqueeze(0).to(device)
     model.eval()
     generated_ids = input_ids.tolist()[0]