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 |
|