chansung sayakpaul HF staff commited on
Commit
78f1305
1 Parent(s): 3a69622

Update app.py (#2)

Browse files

- Update app.py (8122051678094a8d979ed15f791ba98ffbe5d28d)
- Upload ADE_val_00000001.jpeg (70270c6bd74d73a80cffb771d18e8720c04da340)


Co-authored-by: Sayak Paul <sayakpaul@users.noreply.huggingface.co>

Files changed (2) hide show
  1. ADE_val_00000001.jpeg +0 -0
  2. app.py +2 -2
ADE_val_00000001.jpeg ADDED
app.py CHANGED
@@ -9,7 +9,7 @@ feature_extractor = SegformerFeatureExtractor.from_pretrained(
9
  "nvidia/segformer-b5-finetuned-ade-640-640"
10
  )
11
  model = TFSegformerForSemanticSegmentation.from_pretrained(
12
- "nvidia/segformer-b5-finetuned-ade-640-640", from_pt=True
13
  )
14
 
15
  def ade_palette():
@@ -196,6 +196,6 @@ def sepia(input_img):
196
  pred_img = pred_img.astype(np.uint8)
197
  return pred_img
198
 
199
- demo = gr.Interface(sepia, gr.Image(shape=(200, 200)), "image")
200
 
201
  demo.launch()
 
9
  "nvidia/segformer-b5-finetuned-ade-640-640"
10
  )
11
  model = TFSegformerForSemanticSegmentation.from_pretrained(
12
+ "nvidia/segformer-b5-finetuned-ade-640-640"
13
  )
14
 
15
  def ade_palette():
 
196
  pred_img = pred_img.astype(np.uint8)
197
  return pred_img
198
 
199
+ demo = gr.Interface(sepia, gr.Image(shape=(200, 200)), "image", examples=["ADE_val_00000001.jpeg"])
200
 
201
  demo.launch()