Update app.py

app.py

@@ -13,12 +13,16 @@ df['real'] = df['real'].apply(lambda x: f"{x:.2f}")
 # Fill NaN values and convert all columns to strings
 df = df.fillna('').astype(str)
 
-# Function to generate a response using the TAPAS model
+# Ensure that the 'Group' column doesn't have excessive length or unexpected values
+df['Group'] = df['Group'].str.slice(0, 255)  # Truncate to 255 characters if needed
+
+# Function to generate a response using the TAPEX model
 def response(user_question, df):
     a = datetime.datetime.now()
 
     # Initialize the TAPAS model
-    tqa = pipeline(task="table-question-answering", model="google/tapas-large-finetuned-wtq")
+    tqa = pipeline(task="table-question-answering", model="google/tapas-large-finetuned-wtq", 
+                   tokenizer_kwargs={"clean_up_tokenization_spaces": False})
 
     # Debugging information
     print("DataFrame shape:", df.shape)
@@ -26,8 +30,12 @@ def response(user_question, df):
     print("User question:", user_question)
 
     # Query the TAPAS model
-    answer = tqa(table=df, query=user_question)['answer']
-
+    try:
+        answer = tqa(table=df, query=user_question)['answer']
+    except IndexError as e:
+        print(f"Error: {e}")
+        answer = "Error occurred: " + str(e)
+    
     query_result = {
         "Resposta": answer
     }