Update app.py

app.py

@@ -1,27 +1,17 @@
 import gradio as gr
-from transformers import pipeline
 
-# Load the QA model only once at the start
-qa_pipeline = pipeline("question-answering", model="deepset/roberta-base-squad2")
-
-def get_answer(question):
-    # Predefined factual context
-    context = """
-    Washington, D.C. is the capital of the United States of America.
-    New Delhi is the capital of India.
-    London is the capital of the United Kingdom.
-    """
-    
-    answer = qa_pipeline(question=question, context=context)
-    return answer["answer"]
-
-# Create Gradio Interface
+# Define the Gradio interface
 iface = gr.Interface(
-    fn=get_answer,
-    inputs="text",
-    outputs="text",
-    title="Ask Any Question",
-    description="Ask factual questions and get precise answers."
+    fn=generate_response,
+    inputs=[
+        gr.inputs.Textbox(lines=2, label="Input Prompt"),
+        gr.inputs.Slider(minimum=50, maximum=200, step=10, default=100, label="Max Length"),
+        gr.inputs.Slider(minimum=0.1, maximum=1.0, step=0.1, default=0.7, label="Temperature")
+    ],
+    outputs=gr.outputs.Textbox(label="Generated Response"),
+    title="Llama 2 Chatbot",
+    description="Interact with the Llama 2 model using Gradio."
 )
 
+# Launch the interface
 iface.launch()