Update app.py

app.py

@@ -45,11 +45,13 @@ 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({"class_name": class_name, "coordinates": (x, y)})
+        extracted_tags.append(class_name)  # Only appending the class_name
         draw.text((x, y), class_name, fill='black')
 
     return image_mask, extracted_tags
 
+
+
 def gradio_process(image):
     inputs = processor(images=image, return_tensors="pt")
 
@@ -74,10 +76,11 @@ model.eval()
 demo = gr.Interface(
     gradio_process, 
     inputs=gr.inputs.Image(type="pil"), 
-    outputs=[gr.outputs.Image(type="pil"), gr.outputs.JSON()],
+    outputs=[gr.outputs.Image(type="pil"), gr.outputs.Text()],
     title="Semantic Segmentation",
     examples=glob.glob('./examples/*.jpg'),
     allow_flagging="never",
 )
 
+
 demo.launch()