Update app.py

app.py

@@ -1,29 +1,19 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
 
-# Initialize the InferenceClient
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
-# Define the function for generating output based on a prompt
-def generate_output():
-    # Initialize the prompt
-    prompt = "Initialize your prompt here."
-    # Generate output based on the prompt
+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']
 
-# Define the action to be taken when the button is clicked
-def button_click():
-    # Generate output based on the prompt when the button is clicked
-    output_textbox.update(generate_output())
+demo = gr.Interface(
+    fn=get_output,
+    inputs="text",
+    outputs="text",
+    title="LLaMA Gradio App",
+    description="Click the button to get the output",
+    button="Get Output"
+)
 
-# Create a Gradio interface with a button and a text output
-button = gr.Button("Generate Output", onclick=button_click)
-output_textbox = gr.outputs.Textbox(label="Output")
-
-# Combine the button and the text output in a Gradio interface
-interface = gr.Interface(button, output_textbox)
-
-# Launch the app
-if __name__ == "__main__":
-    interface.launch()
+demo.launch()