Update app.py

app.py

@@ -7,13 +7,16 @@ from llama_index.embeddings.langchain import LangchainEmbedding
 from llama_index.core import PromptHelper
 from typing import Optional, List, Mapping, Any
 import pandas as pd
-
+# Load model directly
+from transformers import AutoModel
 # Set the page config as the first command
 st.set_page_config(page_title='Mental Heallth chatbot', page_icon=':robot_face:', layout='wide')
 
 # Define constants
 MODEL_NAME = 'TheBloke/MelloGPT-GGUF'
-MODEL_PATH = 'TheBloke/MelloGPT-GGUF'
+MODEL_PATH = AutoModel.from_pretrained("TheBloke/MelloGPT-GGUF")
+
+#MODEL_PATH = 'TheBloke/MelloGPT-GGUF'
 KNOWLEDGE_BASE_FILE = "mentalhealth.csv"
 
 # Configuration