Update app.py

app.py

@@ -239,19 +239,14 @@ class RAGPipeline:
                 "verbose": False
             }
         
-            return Llama(model_path=self.model_path, **llm_config)
-        
-        except FileNotFoundError as e:
-            logging.error(f"Failed to find or download model file: {str(e)}")
-            raise
-        
-        except ValueError as e:
-            logging.error(f"Model file validation failed: {str(e)}")
-            raise
+            self.llm = Llama(model_path=self.model_path, 
+                             **llm_config)
         
+            st.success("Model loaded successfully!")
+
         except Exception as e:
-            logging.error(f"Unexpected error during model initialization: {str(e)}")
-            raise RuntimeError(f"Failed to initialize model: {str(e)}") from e
+            st.error(f"Error initializing model: {str(e)}")
+            raise
 
         
     @log_function