mukesh3444 commited on
Commit
089df3a
1 Parent(s): 48d2260

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -20,9 +20,9 @@ os.environ['SENTENCE_TRANSFORMERS_HOME'] = '/code/.cache'
20
  os.environ['TRANSFORMERS_CACHE '] = '/code/.cache'
21
 
22
  device = torch.device("cpu")
23
- model = MaskFormerForInstanceSegmentation.from_pretrained("maskformer-swin-large-ade").to(device)
24
  model.eval()
25
- preprocessor = MaskFormerFeatureExtractor.from_pretrained("maskformer-swin-large-ade")
26
 
27
  class LABEL_TYPE(Enum):
28
  WINDOW = 8
 
20
  os.environ['TRANSFORMERS_CACHE '] = '/code/.cache'
21
 
22
  device = torch.device("cpu")
23
+ model = MaskFormerForInstanceSegmentation.from_pretrained("facebook/maskformer-swin-large-ade").to(device)
24
  model.eval()
25
+ preprocessor = MaskFormerFeatureExtractor.from_pretrained("facebook/maskformer-swin-large-ade")
26
 
27
  class LABEL_TYPE(Enum):
28
  WINDOW = 8