Spaces:
Sleeping
Sleeping
Update agent.py
Browse files
agent.py
CHANGED
|
@@ -12,7 +12,7 @@ class GaiaAgent:
|
|
| 12 |
self.model = InferenceClientModel(
|
| 13 |
max_tokens = 1024,
|
| 14 |
temperature = 0.5,
|
| 15 |
-
model_id =
|
| 16 |
#custom_role_conversions=None,
|
| 17 |
provider="hf-inference"
|
| 18 |
)
|
|
|
|
| 12 |
self.model = InferenceClientModel(
|
| 13 |
max_tokens = 1024,
|
| 14 |
temperature = 0.5,
|
| 15 |
+
model_id = 'https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud', #"meta-llama/Llama-3.1-8B-Instruct",
|
| 16 |
#custom_role_conversions=None,
|
| 17 |
provider="hf-inference"
|
| 18 |
)
|