Update Hugging Face model repository in agent.py to use DeepSeek-R1-0528 for improved text generation.
Browse files- __pycache__/agent.cpython-39.pyc +0 -0
- agent.py +2 -1
__pycache__/agent.cpython-39.pyc
CHANGED
|
Binary files a/__pycache__/agent.cpython-39.pyc and b/__pycache__/agent.cpython-39.pyc differ
|
|
|
agent.py
CHANGED
|
@@ -13,7 +13,8 @@ load_dotenv()
|
|
| 13 |
|
| 14 |
# Base Hugging Face LLM used by the chat wrapper
|
| 15 |
base_llm = HuggingFaceEndpoint(
|
| 16 |
-
repo_id="
|
|
|
|
| 17 |
task="text-generation",
|
| 18 |
temperature=0.0,
|
| 19 |
huggingfacehub_api_token=os.getenv("HUGGINGFACEHUB_API_TOKEN"),
|
|
|
|
| 13 |
|
| 14 |
# Base Hugging Face LLM used by the chat wrapper
|
| 15 |
base_llm = HuggingFaceEndpoint(
|
| 16 |
+
repo_id="deepseek-ai/DeepSeek-R1-0528:together",
|
| 17 |
+
# deepseek-ai/DeepSeek-OCR:novita
|
| 18 |
task="text-generation",
|
| 19 |
temperature=0.0,
|
| 20 |
huggingfacehub_api_token=os.getenv("HUGGINGFACEHUB_API_TOKEN"),
|