Upload app.py

app.py

@@ -31,20 +31,17 @@ examples = [
 
 # Create Gradio UI
 # Launch Gradio app with a submit button
-txt = gr.Textbox(lines=2, placeholder="Enter your prompt here...")
-btn = gr.Button("Submit")
-
-app = gr.Interface(
-    fn=generate_response,
-    inputs=txt,
-    outputs=gr.Textbox(),
-    title="Phi-2 LoRA Model",
-    description="A fine-tuned Phi-2 model with LoRA running on Hugging Face Spaces (CPU optimized).",
-    examples=[[ex] for ex in examples],
-    allow_flagging="never"
-)
-
-btn.click(fn=generate_response, inputs=txt, outputs=gr.Textbox())
+# Create Gradio UI with a submit button
+with gr.Blocks() as app:
+    gr.Markdown("## Phi-2 LoRA Model")
+    gr.Markdown("A fine-tuned Phi-2 model with LoRA running on Hugging Face Spaces (CPU optimized).")
+    txt = gr.Textbox(lines=2, placeholder="Enter your prompt here...")
+    btn = gr.Button("Submit")
+    output = gr.Textbox()
+    
+    btn.click(fn=generate_response, inputs=txt, outputs=output)
+    
+    gr.Examples(examples=[[ex] for ex in examples], inputs=txt)
 
 # Launch Gradio app
 app.launch()