yunnn426 commited on
Commit
ae0b01a
·
1 Parent(s): 1361907

delete mouse hover, add gr.Label

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -6,7 +6,6 @@ import numpy as np
6
  from PIL import Image
7
  import tensorflow as tf
8
  from transformers import SegformerFeatureExtractor, TFSegformerForSemanticSegmentation
9
- import mplcursors # mouse hover
10
 
11
  feature_extractor = SegformerFeatureExtractor.from_pretrained(
12
  "nielsr/segformer-b0-finetuned-segments-sidewalk",
@@ -117,15 +116,14 @@ def sepia(input_img):
117
  pred_img = np.array(input_img) * 0.5 + color_seg * 0.5
118
  pred_img = pred_img.astype(np.uint8)
119
 
120
- fig, ax = draw_plot(pred_img, seg)
121
- mplcursors.cursor(hover=True, ax=ax)
122
  return fig
123
 
124
  demo = gr.Interface(fn=sepia,
125
  title="🚥Sidewalk Segmentation🏃‍️",
126
  description="Image Segmentation for Sidewalks",
127
  inputs=gr.Image(),
128
- outputs=['plot'],
129
  examples=["sidewalk_1.jpg", "sidewalk_2.jpg", "sidewalk_3.jpg", "sidewalk_4.jpg", "sidewalk_5.jpg"],
130
  allow_flagging='never',
131
  theme="gradio/soft",
 
6
  from PIL import Image
7
  import tensorflow as tf
8
  from transformers import SegformerFeatureExtractor, TFSegformerForSemanticSegmentation
 
9
 
10
  feature_extractor = SegformerFeatureExtractor.from_pretrained(
11
  "nielsr/segformer-b0-finetuned-segments-sidewalk",
 
116
  pred_img = np.array(input_img) * 0.5 + color_seg * 0.5
117
  pred_img = pred_img.astype(np.uint8)
118
 
119
+ fig = draw_plot(pred_img, seg)
 
120
  return fig
121
 
122
  demo = gr.Interface(fn=sepia,
123
  title="🚥Sidewalk Segmentation🏃‍️",
124
  description="Image Segmentation for Sidewalks",
125
  inputs=gr.Image(),
126
+ outputs=['plot', gr.Label(label="labels")],
127
  examples=["sidewalk_1.jpg", "sidewalk_2.jpg", "sidewalk_3.jpg", "sidewalk_4.jpg", "sidewalk_5.jpg"],
128
  allow_flagging='never',
129
  theme="gradio/soft",