Hugging Face's logo

Spaces:
gatesla
/
Image-Segmentation-Various
Running

App Files Community
gatesla commited on
Commit
b444b25
·
verified ·
1 Parent(s): 3a9ef72

Forgot 2 imports

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -6,9 +6,10 @@ import torch
6
  import pathlib
7
  from PIL import Image
8
 
9
- from transformers import DetrFeatureExtractor, DetrForSegmentation
10
  from transformers.models.detr.feature_extraction_detr import rgb_to_id
11
 
 
12
  import os
13
 
14
 
 
6
  import pathlib
7
  from PIL import Image
8
 
9
+ from transformers import DetrFeatureExtractor, DetrForSegmentation, MaskFormerImageProcessor, MaskFormerForInstanceSegmentation
10
  from transformers.models.detr.feature_extraction_detr import rgb_to_id
11
 
12
+
13
  import os
14
 
15