Hugging Face's logo

Spaces:
ockkjs
/
Gradio2
Sleeping

App Files Community
ockkjs commited on
Commit
79fdb85
·
1 Parent(s): c8fb538

"change model"

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-b5-finetuned-ade-640-640"
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 = "mattmdjaga/segformer_b2_clothes"
10
  processor = AutoImageProcessor.from_pretrained(MODEL_ID)
11
  model = AutoModelForSemanticSegmentation.from_pretrained(MODEL_ID)
12