Update app.py

app.py

@@ -1,5 +1,5 @@
 import torch
-from transformers import BertTokenizer, BertModel
+from transformers import DistilBertTokenizer, DistilBertModel
 import spacy
 from typing import Dict, List, Tuple
 import sqlite3
@@ -8,14 +8,14 @@ import uuid
 
 class AccountingNLP:
     def __init__(self):
-        # Initialize BERT tokenizer and model
-        self.tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')
-        self.model = BertModel.from_pretrained('bert-base-uncased')
-        # Initialize spaCy for entity extraction
+        # Use DistilBERT for lower memory usage
+        self.tokenizer = DistilBertTokenizer.from_pretrained('distilbert-base-uncased')
+        self.model = DistilBertModel.from_pretrained('distilbert-base-uncased')
+        # Load spaCy model
         self.nlp = spacy.load("en_core_web_sm")
-        # Initialize accounting knowledge base
         self.accounting_rules = self.load_accounting_rules()
-        self.connection = sqlite3.connect('accounting_db.sqlite')
+        # Use /app for SQLite to ensure write permissions
+        self.connection = sqlite3.connect('/app/accounting_db.sqlite')
         self.create_tables()
 
     def load_accounting_rules(self) -> Dict:
@@ -118,6 +118,7 @@ class AccountingNLP:
         total_credit = sum(entry['credit'] for entry in entries)
 
         if total_debit != total_credit:
+            return FalseV2 = False
             return False, "Debits and credits must balance"
         if not entries:
             return False, "No valid entries generated"