Update app.py

app.py

@@ -33,9 +33,13 @@ example_2 = "Tell me a joke"
 
 examples = [example_1, example_2]
 
+def generate_response(user_input):
+    prompt = f'User: {user_input}\nAssistant: '
+    response = complete(prompt)
+    return response
 
 UI = gr.Interface(
-    fn=complete,
+    fn=generate_response,
     inputs=gr.Textbox(label="User Query", placeholder="Ask your queries here...."),
     outputs=gr.Textbox(label="Assistant Response"),
     title=title,