Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -71,10 +71,13 @@ class WikipediaTool(Tool):
|
|
| 71 |
|
| 72 |
class BasicAgent:
|
| 73 |
def __init__(self):
|
|
|
|
|
|
|
| 74 |
self.model = HfApiModel(
|
| 75 |
model_id='HuggingFaceH4/zephyr-7b-beta',
|
| 76 |
max_tokens=2096,
|
| 77 |
-
temperature=0.3
|
|
|
|
| 78 |
)
|
| 79 |
#HF_TOKEN = os.getenv("AGENT_TOKEN")
|
| 80 |
#self.model = HfApiModel(
|
|
|
|
| 71 |
|
| 72 |
class BasicAgent:
|
| 73 |
def __init__(self):
|
| 74 |
+
|
| 75 |
+
HF_TOKEN = os.getenv("AGENT_TOKEN")
|
| 76 |
self.model = HfApiModel(
|
| 77 |
model_id='HuggingFaceH4/zephyr-7b-beta',
|
| 78 |
max_tokens=2096,
|
| 79 |
+
temperature=0.3,
|
| 80 |
+
token=HF_TOKEN
|
| 81 |
)
|
| 82 |
#HF_TOKEN = os.getenv("AGENT_TOKEN")
|
| 83 |
#self.model = HfApiModel(
|