Update app.py

app.py

@@ -9,6 +9,7 @@ from langchain.chains import RetrievalQA
 from langchain.chat_models import ChatOpenAI
 from typing import List
 from together import Together
+from transformers import AutoTokenizer, AutoModelForCausalLM
 
 
 import streamlit as st
@@ -104,8 +105,8 @@ st.markdown("""
 
 class HuggingFaceEmbeddings(Embeddings):
     def __init__(self, model_name: str):
-        self.tokenizer = AutoTokenizer.from_pretrained(model_name)
-        self.model = AutoModel.from_pretrained(model_name)
+        self.tokenizer = AutoTokenizer.from_pretrained("deepseek-ai/DeepSeek-R1", trust_remote_code=True)
+        self.model = AutoModelForCausalLM.from_pretrained("deepseek-ai/DeepSeek-R1", trust_remote_code=True)
 
     def embed_documents(self, texts: List[str]) -> List[List[float]]:
         embeddings = []