Update app.py

app.py

@@ -24,14 +24,14 @@ def response(user_question, table_data):
     ans = tokenizer.batch_decode(outputs, skip_special_tokens=True)
 
     query_result = {
-        "query": query,
+        "query": user_question,
         "answer": ans[0]
     }
 
     b = datetime.datetime.now()
     print(b - a)
 
-    return query_result, table
+    return query_result
 
 # Streamlit interface
 st.markdown("""