Update app.py
Browse files
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
|
| 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 =
|
| 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 |
)
|