Update main.py

main.py

@@ -30,14 +30,20 @@ class PDFProcessor:
         self.mineru = Mineru()
         self.emb_model = SentenceTransformer('all-MiniLM-L6-v2')
         
-        # Initialize quantized LLM (using deepseek-1.3b)
-        self.llm = Llama(
-            model_path="models/deepseek-1.3b-q5_k_m.gguf",
+        # Initialize LLM with automatic download
+        self.llm = self._initialize_llm()
+        
+    def _initialize_llm(self):
+        """Initialize LLM with automatic download if needed"""
+        return Llama.from_pretrained(
+            repo_id="TheBloke/deepseek-1.3B-GGUF",
+            filename="deepseek-1.3b.Q5_K_M.gguf",
             n_ctx=2048,
             n_threads=os.cpu_count() - 1,
-            n_gpu_layers=35 if os.getenv('USE_GPU') else 0
+            n_gpu_layers=35 if os.getenv('USE_GPU') else 0,
+            verbose=False
         )
-        
+    
     def extract_layout(self, pdf_path: str) -> List[Layout]:
         """Extract structured layout using MinerU"""
         return self.mineru.process_pdf(pdf_path)