Hugging Face's logo

Spaces:
pastenishian
/
6th_mk2
Sleeping

App Files Community
pastenishian commited on
Commit
82a29ba
·
1 Parent(s): 7c8267d

fix: upgrade b3

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ from PIL import Image
6
  import torch
7
  from transformers import AutoImageProcessor, AutoModelForSemanticSegmentation
8
 
9
- MODEL_ID = "nvidia/segformer-b2-finetuned-cityscapes-1024-1024"
10
  processor = AutoImageProcessor.from_pretrained(MODEL_ID)
11
  model = AutoModelForSemanticSegmentation.from_pretrained(MODEL_ID)
12
 
 
6
  import torch
7
  from transformers import AutoImageProcessor, AutoModelForSemanticSegmentation
8
 
9
+ MODEL_ID = "nvidia/segformer-b3-finetuned-cityscapes-1024-1024"
10
  processor = AutoImageProcessor.from_pretrained(MODEL_ID)
11
  model = AutoModelForSemanticSegmentation.from_pretrained(MODEL_ID)
12