sayakpaul HF staff commited on
Commit
8122051
1 Parent(s): 3a69622

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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()