Update app.py

app.py

@@ -66,7 +66,7 @@ sql_model = BartForConditionalGeneration.from_pretrained(sql_model_name)
 #sql_response = None
 conversation_history = []
 
-def predict(input, conversation_history): #history=[]):
+def predict(input): #history=[]):
 
     #global sql_response
     # Check if the user input is a question
@@ -109,9 +109,9 @@ def chatbot_interface(user_input):
     conversation_history.append(user_input)
 
     dialog_prompt = "User: " + " ".join(conversation_history) + "\nBot:"
-    response = predict(dialog_prompt, conversation_history)
+    response = predict(dialog_prompt)
     conversation_history.append(response)
-    return "Bot (DialoGPT): " + response
+    return response
     
 def sqlquery(input):
 
@@ -124,8 +124,8 @@ def sqlquery(input):
 
 chat_interface = gr.Interface(
     fn=chatbot_interface, 
-    inputs=["text"],
-    outputs=["chatbot"], 
+    inputs=gr.Textbox(prompt="You:"),
+    outputs=gr.Textbox(),
     live=True,
     
 )