Update app.py

app.py

@@ -8,17 +8,17 @@ import pandas as pd
 import torch
 
 # Load Hugging Face token from environment
-hf_token = os.getenv("HF_TOKEN")  # Set this in Space Secrets [[2]]
+hf_token = os.getenv("HF_TOKEN")  # Set this in Space Secrets 
 
-# Load Mistral-7B-Instruct with authentication
+# Load Mistral-7B-Instruct with authentication and fast tokenizer
 model_name = "mistralai/Mistral-7B-Instruct-v0.3"
 
 try:
-    tokenizer = AutoTokenizer.from_pretrained(model_name, token=hf_token)
+    tokenizer = AutoTokenizer.from_pretrained(model_name, token=hf_token, use_fast=True)
     model = AutoModelForCausalLM.from_pretrained(
         model_name, 
         torch_dtype=torch.float16, 
-        token=hf_token  # Pass token for gated repo access [[6]]
+        token=hf_token
     )
     pipe = pipeline("text-generation", model=model, tokenizer=tokenizer, max_new_tokens=500)
 except Exception as e:
@@ -47,8 +47,10 @@ def parse_bank_statement(text):
     - Debit / Credit
     - Closing Balance
     - Expense Type (if available)
+
     Return the results in JSON format with keys: 
     ["date", "description", "amount", "debit_credit", "closing_balance", "expense_type"].
+
     Example:
     {{
       "transactions": [
@@ -62,6 +64,7 @@ def parse_bank_statement(text):
         }}
       ]
     }}
+
     Bank Statement Text:
     {text}
     """