Update app.py

app.py

@@ -24,10 +24,10 @@ docs = text_splitter.create_documents([sample_logs])
 embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
 db = FAISS.from_documents(docs, embeddings)
 
-# ✅ Load Codestral (Mistral AI, tuned for code + RCA)
-model_id = "mistralai/codestral"
-tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
-model = AutoModelForCausalLM.from_pretrained(model_id, trust_remote_code=True)
+# ✅ Use DeepSeek Coder (fast, public, code-tuned)
+model_id = "deepseek-ai/deepseek-coder-1.3b-instruct"
+tokenizer = AutoTokenizer.from_pretrained(model_id)
+model = AutoModelForCausalLM.from_pretrained(model_id)
 llm_pipeline = pipeline("text-generation", model=model, tokenizer=tokenizer, max_new_tokens=256)
 llm = HuggingFacePipeline(pipeline=llm_pipeline)