Update app.py

app.py

@@ -68,14 +68,14 @@ def gen_pred(img, model=seg_model):
     nparr = np.frombuffer(img_bytes, np.byte)
     pred_pil = cv2.imdecode(nparr, cv2.IMREAD_REDUCED_COLOR_8)
     # return "UI in developing process ..."
-    return pred
+    return pred_pil
 
 title = "<h1 style='text-align: center;'>Semantic Segmentation</h1>"
 description = "Upload an image and get prediction mask"
 # css_code='body{background-image:url("file=wave.mp4");}'
 
 gr.Interface(fn=gen_pred, 
-             inputs=[gr.Image()], 
+             inputs=[gr.Image(), seg_model], 
              outputs='image', 
              title=title, 
              examples=[["003e2c95d.jpg"], ["003b50a15.jpg"], ["003b48a9e.jpg"], ["0038cbe45.jpg"], ["00371aa92.jpg"]],