Spaces:
Runtime error
Runtime error
fixed coco
Browse files
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/
|
| 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)
|