Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -71,14 +71,18 @@ class WikipediaTool(Tool):
|
|
| 71 |
|
| 72 |
class BasicAgent:
|
| 73 |
def __init__(self):
|
| 74 |
-
HF_TOKEN = os.getenv("AGENT_TOKEN")
|
| 75 |
self.model = HfApiModel(
|
|
|
|
| 76 |
max_tokens=2096,
|
| 77 |
-
temperature=0.3
|
| 78 |
-
model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
|
| 79 |
-
custom_role_conversions=None,
|
| 80 |
-
token=HF_TOKEN
|
| 81 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 82 |
|
| 83 |
|
| 84 |
self.tools = [
|
|
|
|
| 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(
|
| 81 |
+
# max_tokens=2096,
|
| 82 |
+
# temperature=0.3,
|
| 83 |
+
##custom_role_conversions=None,
|
| 84 |
+
#token=HF_TOKEN
|
| 85 |
+
#)
|
| 86 |
|
| 87 |
|
| 88 |
self.tools = [
|