Update main.py

main.py

@@ -6,8 +6,8 @@ from pathlib import Path
 from typing import List, Dict, Optional
 from dataclasses import dataclass
 from fastapi.encoders import jsonable_encoder
-from sentence_transformers import SentenceTransformer
-from llama_cpp import Llama
+# from sentence_transformers import SentenceTransformer
+# from llama_cpp import Llama
 
 # Fix: Dynamically adjust the module path if magic_pdf is in a non-standard location
 try:
@@ -42,17 +42,17 @@ class PDFProcessor:
         self.output_dir.mkdir(exist_ok=True)
 
     def _initialize_emb_model(self, model_name):
-        try:
-            model = SentenceTransformer("sentence-transformers/" + model_name)
-            model.save('models/'+ model_name)
-            return model
-        except:
-            # Load model directly
-            from transformers import AutoTokenizer, AutoModel
+        # try:
+        #     model = SentenceTransformer("sentence-transformers/" + model_name)
+        #     model.save('models/'+ model_name)
+        #     return model
+        # except:
+        # Load model directly
+        from transformers import AutoTokenizer, AutoModel
 
-            tokenizer = AutoTokenizer.from_pretrained("sentence-transformers/all-MiniLM-L6-v2")
-            model = AutoModel.from_pretrained("sentence-transformers/all-MiniLM-L6-v2")
-            return model
+        tokenizer = AutoTokenizer.from_pretrained("sentence-transformers/all-MiniLM-L6-v2")
+        model = AutoModel.from_pretrained("sentence-transformers/all-MiniLM-L6-v2")
+        return model
 
     def _initialize_llm(self, model_name):
         """Initialize LLM with automatic download if needed"""