Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
89adc00
1
Parent(s):
b142874
debug
Browse files- model/models/__init__.py +1 -0
model/models/__init__.py
CHANGED
|
@@ -36,6 +36,7 @@ def load_pipeline(model_name):
|
|
| 36 |
#model_source, model_name, model_type = model_name.split("_")
|
| 37 |
model_source, *rest = model_name.split("_", 1)
|
| 38 |
model_type, model_name = rest[-1], "_".join(rest[:-1])
|
|
|
|
| 39 |
if model_source == "imagenhub":
|
| 40 |
pipe = load_imagenhub_model(model_name, model_type)
|
| 41 |
elif model_source == "playground":
|
|
|
|
| 36 |
#model_source, model_name, model_type = model_name.split("_")
|
| 37 |
model_source, *rest = model_name.split("_", 1)
|
| 38 |
model_type, model_name = rest[-1], "_".join(rest[:-1])
|
| 39 |
+
print("model_source, model_name, model_type =", model_source, model_name, model_type)
|
| 40 |
if model_source == "imagenhub":
|
| 41 |
pipe = load_imagenhub_model(model_name, model_type)
|
| 42 |
elif model_source == "playground":
|