Update app.py

app.py

@@ -7,12 +7,10 @@ def get_output(prompt):
     response = client.text_generation(prompt, max_new_tokens=512, temperature=0.7, top_p=0.95)
     return response[0]['generated_text']
 
-demo = gr.Interface(
-    fn=get_output,
-    inputs=["text", "button"],
-    outputs="text",
-    title="LLaMA Gradio App",
-    description="Click the button to get the output"
-)
+with gr.Blocks() as demo:
+    prompt = gr.Textbox(label="Enter your prompt")
+    button = gr.Button("Generate Output")
+    output = gr.Textbox(label="Output")
+    button.click(get_output, inputs=prompt, outputs=output)
 
 demo.launch()