changed transformer version

main.py

@@ -2,8 +2,8 @@
 import asyncio
 import importlib
 from voice.speech_to_text import SpeechToText
-# from voice.intent_classifier import IntentClassifier
-from voice.classifier import TextClassifier
+from voice.intent_classifier import IntentClassifier
+# from voice.classifier import TextClassifier
 from api.endpoints import FMPEndpoints
 from rag.retriever import Retriever
 from rag.sql_db import SQL_Key_Pair
@@ -12,8 +12,8 @@ from rag.web_search import duckduckgo_web_search
 async def process_query(vosk_model_path, audio_data=None, query_text=None, use_retriever=False):
     # Step 1: Initialize components
     stt = SpeechToText(model_path=vosk_model_path)
-    # classifier = IntentClassifier()
-    classifier = TextClassifier()
+    classifier = IntentClassifier()
+    # classifier = TextClassifier()
     endpoints = FMPEndpoints()
     # initialize rag tools
     retriever = Retriever(file_path="./data/financial_data.csv")

voice/classifier.py

@@ -8,7 +8,7 @@ from difflib import SequenceMatcher
 class TextClassifier:
     def __init__(self):
         # Use a larger model for better NER (optional)
-        self.nlp = spacy.load("en_core_web_sm")  # "en_core_web_lg"
+        self.nlp = spacy.load("en_core_web_lg")  # "en_core_web_lg"
         try:
             # Use a smaller, PyTorch-compatible model for zero-shot classification
             self.classifier = pipeline("zero-shot-classification", model="typeform/distilbert-base-uncased-mnli")