Update app.py

app.py

@@ -47,25 +47,29 @@ def infer(
     
     return generated_text
 
-# Create Gradio interface with API support
-demo = gr.Interface(
-    fn=infer,
-    inputs=[
-        gr.Textbox(label="Prompt"),
-        gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, value=0),
-        gr.Checkbox(label="Randomize Seed", value=True),
-        gr.Slider(label="Max Length", minimum=10, maximum=200, value=100),
-        gr.Slider(label="Temperature", minimum=0.1, maximum=1.0, value=0.7),
-        gr.Slider(label="Top P", minimum=0.1, maximum=1.0, value=0.9)
-    ],
-    outputs=gr.Textbox(label="Recommendation"),
-    title="Task Recommender",
-    description="Generate task recommendations"
-)
+# Create Gradio interface
+with gr.Blocks() as demo:
+    # Prompt input
+    prompt = gr.Textbox(label="Prompt")
+    
+    # Advanced settings
+    with gr.Accordion("Advanced Settings", open=False):
+        seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, value=0)
+        randomize_seed = gr.Checkbox(label="Randomize Seed", value=True)
+        max_length = gr.Slider(label="Max Length", minimum=10, maximum=200, value=100)
+        temperature = gr.Slider(label="Temperature", minimum=0.1, maximum=1.0, value=0.7)
+        top_p = gr.Slider(label="Top P", minimum=0.1, maximum=1.0, value=0.9)
+    
+    # Output
+    output = gr.Textbox(label="Recommendation")
+    
+    # Submit button
+    btn = gr.Button("Generate")
+    btn.click(
+        fn=infer, 
+        inputs=[prompt, seed, randomize_seed, max_length, temperature, top_p],
+        outputs=output
+    )
 
-# Enable API
-demo.launch(
-    enable_queue=True,  # Enable request queuing
-    server_port=7860,
-    api_open=True  # Open API endpoint
-)
+# Launch the demo
+demo.launch(server_name="0.0.0.0", server_port=7860)