Spaces:
Running
Running
A-Mahla
commited on
CHG bill orga (#23)
Browse files
cua2-core/src/cua2_core/services/agent_utils/get_model.py
CHANGED
|
@@ -9,4 +9,4 @@ AVAILABLE_MODELS = [
|
|
| 9 |
|
| 10 |
def get_model(model_id: str) -> Model:
|
| 11 |
"""Get the model"""
|
| 12 |
-
return InferenceClientModel(model_id=model_id)
|
|
|
|
| 9 |
|
| 10 |
def get_model(model_id: str) -> Model:
|
| 11 |
"""Get the model"""
|
| 12 |
+
return InferenceClientModel(bill_to="smolagents", model_id=model_id)
|