Update app.py

app.py

@@ -1,8 +1,8 @@
 import streamlit as st
 import pandas as pd
-from transformers import AutoTokenizer, AutoModelForTableQuestionAnswering
-from datetime import datetime
 import requests
+from datetime import datetime
+from transformers import AutoTokenizer, AutoModelForTableQuestionAnswering
 
 class InterestCalculatorApp:
     def __init__(self):
@@ -106,28 +106,19 @@ class InterestCalculatorApp:
 
     def ask_tapas(self, query, table):
         if isinstance(table, pd.DataFrame):
-            # 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] for row in table.values]
-    
-            # Encode tokens and convert them to PyTorch tensors
-            inputs = self.tokenizer(table=tokenized_table, queries=tokenized_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():
     st.title("Interest Calculation App")
 
     app = InterestCalculatorApp()
 
-    app.download_boe_rates()  # Download base rates first
-
     file_path = st.file_uploader("Upload Invoices File", type=["xlsx"])
 
     if file_path is not None:
@@ -150,5 +141,7 @@ def main():
     if st.button("Calculate Interest"):
         app.calculate_interest()
 
+    app.download_boe_rates()
+
 if __name__ == "__main__":
     main()