Update app.py

app.py

@@ -366,12 +366,12 @@ def inference(image,text):
     plt.axis('off')
     
     plt.tight_layout()
-
-    return plt    
+    plt.savefig('out.png', bbox_inches='tight')
+    return 'out.png'   
 
 title = "Lang-Seg"
 description = "Gradio demo for Language-driven Semantic Segmentation. To use it, simply add your image and text or click one of the examples to load them. Read more at the links below."
 
 article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2201.03546' target='_blank'>Language-driven Semantic Segmentation</a> | <a href='https://github.com/isl-org/lang-seg' target='_blank'>Github Repo</a></p>"
 
-gr.Interface(inference,[gr.inputs.Image(type="pil"),"text"],"plot",title=title,description=description,article=article).launch(enable_queue=True)
+gr.Interface(inference,[gr.inputs.Image(type="pil"),"text"],gr.outputs.Image(type="file"),title=title,description=description,article=article).launch(enable_queue=True)