Update app.py

app.py

@@ -104,11 +104,28 @@ class InterestCalculatorApp:
         except requests.RequestException as e:
             st.error(f"Failed to download rates: {e}")
 
-    def ask_tapas(self, query, table):
-        inputs = self.tokenizer(table, query, return_tensors="pt", padding=True)
-        outputs = self.model(**inputs)
-        predicted_answer = self.tokenizer.decode(outputs.logits.argmax(dim=-1))
-        return predicted_answer
+    def ask_tapas(self, query, df):
+        # Convert the DataFrame to a list of dictionaries
+        table = df.to_dict('records')
+    
+        # Initialize tokenized table
+        tokenized_table = []
+    
+        # Tokenize each row in the table
+        for row in table:
+            tokenized_row = []
+            for cell in row.values():
+                tokenized_row.append(self.tokenizer.tokenize(str(cell)))  # Convert cell to string before tokenizing
+            tokenized_table.append(tokenized_row)
+    
+        # Tokenize the query
+        tokenized_query = self.tokenizer.tokenize(query)
+    
+        # Encode tokens and convert them to PyTorch tensors
+        inputs = self.tokenizer(table=tokenized_table, query=tokenized_query, return_tensors="pt", padding=True)
+    
+        return inputs
+
 
 def main():
     st.title("Interest Calculation App")