Update app.py

app.py

@@ -212,12 +212,15 @@ class RAGPipeline:
         self.retriever = SentenceTransformerRetriever()
         self.documents = []
         self.device = torch.device("cpu")
-        
         self.model_path = "mistral-7b-v0.1.Q4_K_M.gguf"
-        self.llm = initialize_model()
+        # Initialize model in init
+        self.llm = None
+        self.initialize_model()
+
 
     @st.cache_resource
     def initialize_model(self):
+        """Initialize the model with proper error handling and verification"""
         try:
             if not os.path.exists(self.model_path):
                 direct_url = "https://huggingface.co/TheBloke/Mistral-7B-v0.1-GGUF/resolve/main/mistral-7b-v0.1.Q4_K_M.gguf"
@@ -239,9 +242,7 @@ class RAGPipeline:
                 "verbose": False
             }
         
-            self.llm = Llama(model_path=self.model_path, 
-                             **llm_config)
-        
+            self.llm = Llama(model_path=self.model_path, **llm_config)
             st.success("Model loaded successfully!")
 
         except Exception as e: