Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 from transformers import pipeline, AutoModelForCausalLM, AutoTokenizer
 import torch
 import theme
+import chatbot
 
 theme = theme.Theme()
 
@@ -22,11 +23,12 @@ image_gradio_app = gr.Interface(
 
 # Cell 2: Chatbot Model
 
-def echo(message, history):
-    return message
+def qa_response(user_message, chat_history, context):
+    response = qa_chain.predict(user_message, chat_history, context=context)
+    return response
 
 chatbot_gradio_app = gr.ChatInterface(
-    fn=echo,
+    fn=qa_response,
     title="Green Greta",
     theme=theme
 )