Spaces:
Running
Running
Update externalmod.py
Browse files- externalmod.py +1 -1
externalmod.py
CHANGED
@@ -115,7 +115,7 @@ def from_model(model_name: str, hf_token: str | None, alias: str | None, **kwarg
|
|
115 |
|
116 |
headers["X-Wait-For-Model"] = "true"
|
117 |
client = huggingface_hub.InferenceClient(
|
118 |
-
model=model_name, headers=headers, token=hf_token
|
119 |
)
|
120 |
|
121 |
# For tasks that are not yet supported by the InferenceClient
|
|
|
115 |
|
116 |
headers["X-Wait-For-Model"] = "true"
|
117 |
client = huggingface_hub.InferenceClient(
|
118 |
+
model=model_name, headers=headers, token=hf_token, timeout=120,
|
119 |
)
|
120 |
|
121 |
# For tasks that are not yet supported by the InferenceClient
|