Update app.py

app.py

@@ -5,6 +5,7 @@ import pickle
 from PIL import Image, ImageDraw
 import numpy as np
 from transformers import AutoImageProcessor, Mask2FormerForUniversalSegmentation
+
 from scipy.ndimage import center_of_mass
 
 def combine_ims(im1, im2, val=128):
@@ -44,7 +45,7 @@ def get_out_image(image, predicted_semantic_map):
     extracted_tags = []
     for id, (y, x) in class_centers.items():
         class_name = str(class_names[id - 1])
-        extracted_tags.append(f"{class_name}: ({x}, {y})")
+        extracted_tags.append({"class_name": class_name, "coordinates": (x, y)})
         draw.text((x, y), class_name, fill='black')
 
     return image_mask, extracted_tags
@@ -73,7 +74,7 @@ model.eval()
 demo = gr.Interface(
     gradio_process, 
     inputs=gr.inputs.Image(type="pil"), 
-    outputs=[gr.outputs.Image(type="pil"), gr.outputs.Text()],
+    outputs=[gr.outputs.Image(type="pil"), gr.outputs.JSON()],
     title="Semantic Segmentation",
     examples=glob.glob('./examples/*.jpg'),
     allow_flagging="never",