Update app.py

app.py

@@ -106,19 +106,19 @@ class InterestCalculatorApp:
 
     def ask_tapas(self, query, table):
         if isinstance(table, pd.DataFrame):
-            # Convert the DataFrame to a list of dictionaries
-            table_dict = table.to_dict('records')
+            # Tokenize the query
+            tokenized_query = self.tokenizer.tokenize(query)
     
             # Tokenize each cell in the table
-            tokenized_table = [[self.tokenizer.tokenize(str(cell)) for cell in row.values()] for row in table_dict]
+            tokenized_table = [[self.tokenizer.tokenize(str(cell)) for cell in row] for row in table.values]
     
             # Encode tokens and convert them to PyTorch tensors
-            inputs = self.tokenizer(table=tokenized_table, queries=query, return_tensors="pt", padding=True)
+            inputs = self.tokenizer(table=tokenized_table, queries=tokenized_query, return_tensors="pt", padding=True)
     
             return inputs
         else:
             raise TypeError("Table must be of type pd.DataFrame")
-    
+
 
 
 def main():