Aleksandar commited on
Commit
e22a0c0
·
verified ·
1 Parent(s): 82dcc08

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ device = torch.device("cpu")
10
  model = MaskFormerForInstanceSegmentation.from_pretrained("facebook/maskformer-swin-large-coco").to(device)
11
  model.eval()
12
  # preprocessor = MaskFormerFeatureExtractor.from_pretrained("facebook/maskformer-swin-tiny-ade")
13
- model = MaskFormerFeatureExtractor.from_pretrained("facebook/maskformer-swin-large-coco")
14
 
15
  def visualize_instance_seg_mask(mask):
16
  image = np.zeros((mask.shape[0], mask.shape[1], 3))
 
10
  model = MaskFormerForInstanceSegmentation.from_pretrained("facebook/maskformer-swin-large-coco").to(device)
11
  model.eval()
12
  # preprocessor = MaskFormerFeatureExtractor.from_pretrained("facebook/maskformer-swin-tiny-ade")
13
+ preprocessor = MaskFormerFeatureExtractor.from_pretrained("facebook/maskformer-swin-large-coco")
14
 
15
  def visualize_instance_seg_mask(mask):
16
  image = np.zeros((mask.shape[0], mask.shape[1], 3))