mouse hover, theme

app.py

@@ -6,6 +6,7 @@ import numpy as np
 from PIL import Image
 import tensorflow as tf
 from transformers import SegformerFeatureExtractor, TFSegformerForSemanticSegmentation
+import mplcursors # mouse hover
 
 feature_extractor = SegformerFeatureExtractor.from_pretrained(
     "nielsr/segformer-b0-finetuned-segments-sidewalk",
@@ -116,7 +117,8 @@ def sepia(input_img):
     pred_img = np.array(input_img) * 0.5 + color_seg * 0.5
     pred_img = pred_img.astype(np.uint8)
 
-    fig = draw_plot(pred_img, seg)
+    fig, ax = draw_plot(pred_img, seg)
+    mplcursors.cursor(hover=True, ax=ax)
     return fig
 
 demo = gr.Interface(fn=sepia,
@@ -126,7 +128,8 @@ demo = gr.Interface(fn=sepia,
                     outputs=['plot'],
                     examples=["sidewalk_1.jpg", "sidewalk_2.jpg", "sidewalk_3.jpg", "sidewalk_4.jpg", "sidewalk_5.jpg"],
                     allow_flagging='never',
-                    theme="darkhuggingface")
+                    theme="gradio/soft",
+                    live=True)
 
 
 demo.launch()