Update app.py

app.py

@@ -412,10 +412,11 @@ def segment(image):
     plt.subplot(1,2,2)
     plt.imshow(context_rcnn_image)
     plt.title('Context R-CNN')
-    io_buf = io.BytesIO()
-    plt.savefig('out.png', dpi=600)
-    im = Image.open('out.png')
+    img_buf = io.BytesIO()
+    plt.savefig(img_buf, format='png')
+    im = Image.open(img_buf)
+    img_buf.close()
     return im
 examples = os.listdir('../../Examples')
 examples = ['../../Examples/' + item for item in examples]
-gr.Interface(fn=segment, inputs = "file",outputs = "image", examples=examples,css=css,enable_queue=True).launch()
+gr.Interface(fn=segment, inputs = "file",outputs = "image", examples=examples,enable_queue=True).launch()