Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -61,7 +61,7 @@ class GAIAAgent:
|
|
| 61 |
def __init__(self):
|
| 62 |
hf_token = os.environ.get("HF_TOKEN")
|
| 63 |
model_id = os.environ.get("GAIA_MODEL_ID") # optional override
|
| 64 |
-
provider = os.environ.get("GAIA_PROVIDER"
|
| 65 |
|
| 66 |
# Fail fast and loud instead of letting all 20 questions die silently
|
| 67 |
# at Step 1 with a generic "api_key" error from deep inside smolagents.
|
|
@@ -74,7 +74,9 @@ class GAIAAgent:
|
|
| 74 |
"not always hot-reload a running container)."
|
| 75 |
)
|
| 76 |
|
| 77 |
-
model_kwargs = {"token": hf_token
|
|
|
|
|
|
|
| 78 |
if model_id:
|
| 79 |
model_kwargs["model_id"] = model_id
|
| 80 |
model = InferenceClientModel(**model_kwargs)
|
|
|
|
| 61 |
def __init__(self):
|
| 62 |
hf_token = os.environ.get("HF_TOKEN")
|
| 63 |
model_id = os.environ.get("GAIA_MODEL_ID") # optional override
|
| 64 |
+
provider = os.environ.get("GAIA_PROVIDER") # optional override; unset = library default ("auto")
|
| 65 |
|
| 66 |
# Fail fast and loud instead of letting all 20 questions die silently
|
| 67 |
# at Step 1 with a generic "api_key" error from deep inside smolagents.
|
|
|
|
| 74 |
"not always hot-reload a running container)."
|
| 75 |
)
|
| 76 |
|
| 77 |
+
model_kwargs = {"token": hf_token}
|
| 78 |
+
if provider:
|
| 79 |
+
model_kwargs["provider"] = provider
|
| 80 |
if model_id:
|
| 81 |
model_kwargs["model_id"] = model_id
|
| 82 |
model = InferenceClientModel(**model_kwargs)
|