Update app.py
Browse files
app.py
CHANGED
|
@@ -29,13 +29,16 @@ 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=
|
| 34 |
-
|
| 35 |
-
|
| 36 |
-
|
| 37 |
-
|
| 38 |
-
|
|
|
|
| 39 |
)
|
| 40 |
|
| 41 |
retrieval_chain = ConversationalRetrievalChain.from_llm(
|
|
|
|
| 29 |
retriever = vectorstore.as_retriever(search_kwargs={"k": 4})
|
| 30 |
|
| 31 |
# ✅ Wrap DeepSeek model properly
|
| 32 |
+
repo_id = "deepseek-ai/DeepSeek-R1-0528"
|
| 33 |
+
|
| 34 |
llm = HuggingFaceEndpoint(
|
| 35 |
+
repo_id=repo_id,
|
| 36 |
+
max_length=128,
|
| 37 |
+
temperature=0.5,
|
| 38 |
+
provider="auto", # set your provider here hf.co/settings/inference-providers
|
| 39 |
+
# provider="hyperbolic",
|
| 40 |
+
# provider="nebius",
|
| 41 |
+
# provider="together",
|
| 42 |
)
|
| 43 |
|
| 44 |
retrieval_chain = ConversationalRetrievalChain.from_llm(
|