Update app.py

app.py

@@ -9,11 +9,11 @@ df = pd.read_excel('discrepantes.xlsx')
 df.fillna(0, inplace=True)
 
 # Function to generate a response using the TAPEX model
-def response(user_question, table_data):
+def response(user_question, df):
     a = datetime.datetime.now()
 
     tqa = pipeline(task="table-question-answering", model="google/tapas-large-finetuned-wtq")
-    answer = tqa(table=table, query=question)['cells'][0]
+    answer = tqa(table=df, query=user_question)['answer']
     
     query_result = {
         "Resposta": answer