Hugging Face's logo

Spaces:
kmn5409
/
Flower_Demo
Runtime error

App Files Community
kmn5409 commited on
Commit
efcd7d4
·
1 Parent(s): 7801d57

New

Browse files
Files changed (1) hide show
  1. model.py +1 -0
model.py CHANGED
@@ -47,6 +47,7 @@ class Model:
47
  PANOPTIC_SEGMENTATION_MODEL_DICT = _load_model_dict(
48
  'model_dict/panoptic_segmentation.yaml')
49
  MODEL_DICT = DETECTION_MODEL_DICT | INSTANCE_SEGMENTATION_MODEL_DICT | PANOPTIC_SEGMENTATION_MODEL_DICT
 
50
  #MODEL_DICT = INSTANCE_SEGMENTATION_MODEL_DICT
51
 
52
  def __init__(self, model_name: str):
 
47
  PANOPTIC_SEGMENTATION_MODEL_DICT = _load_model_dict(
48
  'model_dict/panoptic_segmentation.yaml')
49
  MODEL_DICT = DETECTION_MODEL_DICT | INSTANCE_SEGMENTATION_MODEL_DICT | PANOPTIC_SEGMENTATION_MODEL_DICT
50
+
51
  #MODEL_DICT = INSTANCE_SEGMENTATION_MODEL_DICT
52
 
53
  def __init__(self, model_name: str):