Update app.py

app.py

@@ -36,7 +36,7 @@ vectordb = FAISS.from_documents(split_docs, embeddings)
 retriever = vectordb.as_retriever(search_kwargs={"k": 5})
 
 # ✅ Step 4: Load Hugging Face Model
-MODEL_NAME = "mistralai/Mistral-7B-Instruct"  # More memory-efficient than Yi-Coder-9B
+MODEL_NAME = "01-ai/Yi-Coder-9B-Chat"  # More memory-efficient than Yi-Coder-9B
 print(f"🚀 Loading Model: {MODEL_NAME} on {device}...")
 
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)