sumit1703 commited on
Commit
991cedd
·
verified ·
1 Parent(s): c489c3a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
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", "hf-inference") # optional override
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, "provider": provider}
 
 
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)