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)
|