xvjiarui commited on
Commit
8799af5
Β·
1 Parent(s): 3095a7d

fixed coco

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -64,7 +64,7 @@ def inference(dataset, additional_classes, input_img):
64
  seg_cfg = 'segmentation/configs/_base_/datasets/pascal_voc12.py'
65
  elif dataset == 'coco' or dataset == 'COCO':
66
  dataset_class = COCOObjectDataset
67
- seg_cfg = 'segmentation/configs/_base_/datasets/coco_object164k.py'
68
  elif dataset == 'context' or dataset == 'Pascal Context':
69
  dataset_class = PascalContextDataset
70
  seg_cfg = 'segmentation/configs/_base_/datasets/pascal_context.py'
@@ -153,4 +153,4 @@ gr.Interface(
153
  title=title,
154
  description=description,
155
  article=article,
156
- examples=examples).launch()
 
64
  seg_cfg = 'segmentation/configs/_base_/datasets/pascal_voc12.py'
65
  elif dataset == 'coco' or dataset == 'COCO':
66
  dataset_class = COCOObjectDataset
67
+ seg_cfg = 'segmentation/configs/_base_/datasets/coco.py'
68
  elif dataset == 'context' or dataset == 'Pascal Context':
69
  dataset_class = PascalContextDataset
70
  seg_cfg = 'segmentation/configs/_base_/datasets/pascal_context.py'
 
153
  title=title,
154
  description=description,
155
  article=article,
156
+ examples=examples).launch(enable_queue=True)