Update app.py

app.py

@@ -101,9 +101,15 @@ def sepia(input_img):
     fig = draw_plot(pred_img, seg)
     return fig
 
+    from io import BytesIO
+    img_byte_array = BytesIO()
+    pred_img.save(img_byte_array, format='PNG')
+    img_data = img_byte_array.getvalue()
+    img_base64 = base64.b64encode(img_data).decode()
+
 demo = gr.Interface(fn=sepia,
                     inputs=gr.Image(shape=(400, 600)),
-                    outputs=['label'],
+                    outputs=['text'],
                     examples=["cityoutdoor-1.jpg", "cityoutdoor-2.jpg", "cityoutdoor-3.jpg"],
                     allow_flagging='never')