Hugging Face's logo

Spaces:
gch0301
/
sixthspace_free
Sleeping

App Files Community
gch0301 commited on
Commit
3212fab
·
1 Parent(s): dffaf20

1

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 = "Xenova/segformer-b0-finetuned-ade-512-512"
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-b4-finetuned-cityscapes-1024-1024"
10
  processor = AutoImageProcessor.from_pretrained(MODEL_ID)
11
  model = AutoModelForSemanticSegmentation.from_pretrained(MODEL_ID)
12