Update app.py

app.py

@@ -15,11 +15,14 @@ def inference(img):
       output_dir='output',
       visualization=True)
   print(result)
-  return result[0][0]
+  return result[0]['front'][:,:,::-1], result[0]['mask']
 
-  
+outputs = [
+           gr.outputs.Image(type="numpy",label="Front"),
+           gr.outputs.Image(type="numpy",label="Mask")
+           ]
 title="u2Net"
 description="U^2-Net: Going Deeper with Nested U-Structure for Salient Object Detection"
 
 examples=[['cat2.jpg']]
-gr.Interface(inference,gr.inputs.Image(type="filepath"),"label",title=title,description=description,examples=examples).launch(enable_queue=True)
+gr.Interface(inference,gr.inputs.Image(type="filepath"),outputs,title=title,description=description,examples=examples).launch(enable_queue=True)