Update multit2i.py
Browse files- multit2i.py +1 -1
multit2i.py
CHANGED
|
@@ -35,7 +35,7 @@ def is_repo_name(s):
|
|
| 35 |
|
| 36 |
def get_status(model_name: str):
|
| 37 |
from huggingface_hub import InferenceClient
|
| 38 |
-
client = InferenceClient(token=HF_TOKEN, timeout=
|
| 39 |
return client.get_model_status(model_name)
|
| 40 |
|
| 41 |
|
|
|
|
| 35 |
|
| 36 |
def get_status(model_name: str):
|
| 37 |
from huggingface_hub import InferenceClient
|
| 38 |
+
client = InferenceClient(token=HF_TOKEN, timeout=150)
|
| 39 |
return client.get_model_status(model_name)
|
| 40 |
|
| 41 |
|