gatesla commited on
Commit
745257f
·
verified ·
1 Parent(s): bdfaed1

Got stuff confused

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,7 +6,7 @@ import torch
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
 
@@ -72,7 +72,7 @@ def segment_images(model_name,url_input,image_input,threshold):
72
  pass
73
  elif "maskformer" in model_name.lower():
74
  # Load the processor and model
75
- processor = MaskFormerForInstanceSegmentation.from_pretrained(model_name)
76
  print(type(processor))
77
  model = MaskFormerForInstanceSegmentation.from_pretrained(model_name)
78
 
 
6
  import pathlib
7
  from PIL import Image
8
 
9
+ from transformers import DetrFeatureExtractor, DetrForSegmentation, MaskFormerFeatureExtractor, MaskFormerForInstanceSegmentation
10
  from transformers.models.detr.feature_extraction_detr import rgb_to_id
11
 
12
 
 
72
  pass
73
  elif "maskformer" in model_name.lower():
74
  # Load the processor and model
75
+ processor = MaskFormerFeatureExtractor.from_pretrained(model_name)
76
  print(type(processor))
77
  model = MaskFormerForInstanceSegmentation.from_pretrained(model_name)
78