vinesmsuic commited on
Commit
89adc00
·
1 Parent(s): b142874
Files changed (1) hide show
  1. 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":