Spaces:
Sleeping
Sleeping
Update agent.py
Browse files
agent.py
CHANGED
|
@@ -17,7 +17,7 @@ class GaiaAgent:
|
|
| 17 |
self.llm = pipeline(
|
| 18 |
"text-generation",
|
| 19 |
model="microsoft/DialoGPT-medium", # <-- Bara denna rad
|
| 20 |
-
|
| 21 |
device="cpu",
|
| 22 |
max_new_tokens=1024,
|
| 23 |
do_sample=False,
|
|
|
|
| 17 |
self.llm = pipeline(
|
| 18 |
"text-generation",
|
| 19 |
model="microsoft/DialoGPT-medium", # <-- Bara denna rad
|
| 20 |
+
token=token,
|
| 21 |
device="cpu",
|
| 22 |
max_new_tokens=1024,
|
| 23 |
do_sample=False,
|