Update app.py

app.py

@@ -218,20 +218,23 @@ class RAGPipeline:
         self.initialize_model()
 
 
-    @st.cache_resource
-    def initialize_model(self):
-        """Initialize the model with proper error handling and verification"""
+    st.cache_resource
+    def initialize_model(_self):
+        """Initialize the model with proper error handling and verification
+        
+        Note: Using _self instead of self for Streamlit caching compatibility
+        """
         try:
-            if not os.path.exists(self.model_path):
+            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"
-                download_file_with_progress(direct_url, self.model_path)
+                download_file_with_progress(direct_url, _self.model_path)
 
             # Verify file exists and has content
-            if not os.path.exists(self.model_path):
-                raise FileNotFoundError(f"Model file {self.model_path} not found after download attempts")
+            if not os.path.exists(_self.model_path):
+                raise FileNotFoundError(f"Model file {_self.model_path} not found after download attempts")
             
-            if os.path.getsize(self.model_path) < 1000000:  # Less than 1MB
-                os.remove(self.model_path)
+            if os.path.getsize(_self.model_path) < 1000000:  # Less than 1MB
+                os.remove(_self.model_path)
                 raise ValueError("Downloaded model file is too small, likely corrupted")
             
             llm_config = {
@@ -242,13 +245,12 @@ 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:
             st.error(f"Error initializing model: {str(e)}")
             raise
-
         
     @log_function
     @st.cache_data