finish

app.py

@@ -105,7 +105,7 @@ def sepia(input_img):
     return fig
 
 
-def compare_images(input_img):
+def with_labels(input_img):
     input_img = Image.fromarray(input_img)
 
     inputs = feature_extractor(images=input_img, return_tensors="tf")
@@ -127,12 +127,12 @@ def compare_images(input_img):
     pred_img = np.array(input_img) * 0.5 + color_seg * 0.5
     pred_img = pred_img.astype(np.uint8)
 
-    return input_img, pred_img
+    return input_img, pred_img, labels_list
 
 
-demo = gr.Interface(fn=compare_images,
+demo = gr.Interface(fn=with_labels,
                     inputs=gr.Image(shape=(1024, 1024)),
-                    outputs=["image", "image"],
+                    outputs=["image", "image", "text"],
                     examples=["city-1.jpg", "city-2.jpg", "city-3.jpg", "city-4.jpg", "city-5.jpg"],
                     allow_flagging='never')