Spaces:
Sleeping
Sleeping
Thomas Bartlett
commited on
Commit
·
50fe614
1
Parent(s):
12fe825
use a better model
Browse files
app.py
CHANGED
|
@@ -63,12 +63,12 @@ class BasicAgent:
|
|
| 63 |
|
| 64 |
self.llm = ChatHuggingFace(
|
| 65 |
llm=HuggingFaceEndpoint(
|
| 66 |
-
repo_id="
|
| 67 |
task="text-generation",
|
| 68 |
max_new_tokens=512,
|
| 69 |
temperature=0.0,
|
| 70 |
do_sample=False,
|
| 71 |
-
provider="
|
| 72 |
)
|
| 73 |
)
|
| 74 |
|
|
|
|
| 63 |
|
| 64 |
self.llm = ChatHuggingFace(
|
| 65 |
llm=HuggingFaceEndpoint(
|
| 66 |
+
repo_id="deepseek-ai/DeepSeek-R1",
|
| 67 |
task="text-generation",
|
| 68 |
max_new_tokens=512,
|
| 69 |
temperature=0.0,
|
| 70 |
do_sample=False,
|
| 71 |
+
provider="auto" # Automatically select the first available provider
|
| 72 |
)
|
| 73 |
)
|
| 74 |
|