Update app.py

app.py

@@ -59,7 +59,7 @@ image_output = gr.outputs.Image(type="pil", label="Output Image")
 def gen_pred(img=inputs, model=seg_model):
     # rgb_path = os.path.join(test_image_dir,img)
     # img = cv2.imread(rgb_path)
-    img = cv2.imread("./003e2c95d.jpg")
+    # img = cv2.imread("./003e2c95d.jpg")
     # pil_image = Image.open('./003b50a15.jpg')
     # img = cv2.cvtColor(np.array(pil_image), cv2.COLOR_RGB2BGR)
     img = img[::IMG_SCALING[0], ::IMG_SCALING[1]]
@@ -72,8 +72,7 @@ def gen_pred(img=inputs, model=seg_model):
     # color_coverted = cv2.cvtColor(pred, cv2.COLOR_BGR2RGB)
     # pil_image = Image.fromarray(pred)
     PIL_image = Image.fromarray(pred.astype('uint8'), 'RGB')
-    "UI in developing process ..."
-    return PIL_image
+    return "UI in developing process ..."
 
 
 title = "<h1 style='text-align: center;'>Semantic Segmentation</h1>"
@@ -82,7 +81,7 @@ description = "Upload an image and get prediction mask"
 
 gr.Interface(fn=gen_pred, 
              inputs=inputs, 
-             outputs="image", 
+             outputs="text", 
              title=title, 
              examples=[["003e2c95d.jpg"], ["003b50a15.jpg"], ["003b48a9e.jpg"], ["0038cbe45.jpg"], ["00371aa92.jpg"]],
              # css=css_code,