Update app.py

app.py

@@ -20,7 +20,7 @@ embeddings = HuggingFaceEmbeddings(model_name="all-MiniLM-L6-v2")
 vectordb = FAISS.from_documents(split_docs, embeddings)
 
 # Load model and create pipeline
-model_name = "01-ai/Yi-Coder-9B-Chat"
+model_name = "01-ai/Yi-Coder-9B"
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 print(f"Using device: {device}")