Bofandra commited on
Commit
73aca0a
·
verified ·
1 Parent(s): 5091cc6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -5,7 +5,7 @@ from langchain.vectorstores import FAISS
5
  from langchain.embeddings import HuggingFaceEmbeddings
6
  from langchain.text_splitter import RecursiveCharacterTextSplitter
7
  from langchain.document_loaders import PyPDFLoader
8
- from langchain_huggingface import HuggingFaceEndpoint
9
  import tempfile
10
 
11
  # Initialize global variables
@@ -29,9 +29,8 @@ def process_pdf(file):
29
  retriever = vectorstore.as_retriever(search_kwargs={"k": 4})
30
 
31
  # ✅ Wrap DeepSeek model properly
32
- llm = HuggingFaceEndpoint(
33
  repo_id="deepseek-ai/DeepSeek-R1-0528",
34
- task="conversational", # 👈 force the correct mode
35
  temperature=0.7,
36
  max_new_tokens=512
37
  )
 
5
  from langchain.embeddings import HuggingFaceEmbeddings
6
  from langchain.text_splitter import RecursiveCharacterTextSplitter
7
  from langchain.document_loaders import PyPDFLoader
8
+ from langchain_huggingface import ChatHuggingFaceEndpoint
9
  import tempfile
10
 
11
  # Initialize global variables
 
29
  retriever = vectorstore.as_retriever(search_kwargs={"k": 4})
30
 
31
  # ✅ Wrap DeepSeek model properly
32
+ llm = ChatHuggingFaceEndpoint(
33
  repo_id="deepseek-ai/DeepSeek-R1-0528",
 
34
  temperature=0.7,
35
  max_new_tokens=512
36
  )