Update app.py

app.py

@@ -412,8 +412,10 @@ def segment(image):
     plt.subplot(1,2,2)
     plt.imshow(context_rcnn_image)
     plt.title('Context R-CNN')
-    return plt
+    io_buf = io.BytesIO()
+    plt.savefig('out.png', dpi=600)
+    im = Image.open('out.png')
+    return im
 examples = os.listdir('../../Examples')
 examples = ['../../Examples/' + item for item in examples]
-css = ".output_image {height: 40rem !important; width: 100% !important;}"
-gr.Interface(fn=segment, inputs = "file",outputs = "plot", examples=examples,css=css,enable_queue=True).launch()
+gr.Interface(fn=segment, inputs = "file",outputs = "image", examples=examples,css=css,enable_queue=True).launch()