Spaces:
Running
on
Zero
Running
on
Zero
vinesmsuic
commited on
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":
|