Update app.py

modify chunking

app.py

@@ -36,8 +36,15 @@ def process_table_query(query, table_data):
         result_tapex = tokenizer_tapex.batch_decode(outputs, skip_special_tokens=True)[0]
         results.append(result_tapex)
 
-    # Aggregate results
-    return ' '.join(results)
+    # Aggregate results (example: summing numerical values if the query is sum-related)
+    try:
+        # Convert the list of results to floats and sum them
+        numerical_results = [float(r) for r in results if r.strip().isdigit()]
+        total_sum = sum(numerical_results)
+        return str(total_sum)
+    except ValueError:
+        # If the results are not numerical, return the joined string
+        return ' '.join(results)
 
 # Gradio interface
 def answer_query_from_csv(query, file):