Update app.py

app.py

@@ -6,9 +6,7 @@ from PIL import Image
 from transparent_background import Remover
 
 # Load model
-remover = Remover() # default setting
-remover = Remover(mode='fast', jit=True, device='cuda:0', ckpt='~/latest.pth') # custom setting
-remover = Remover(mode='base-nightly') # nightly release checkpoint
+remover = Remover(mode='fast') # custom setting
 
 # Usage for image
 
@@ -30,6 +28,7 @@ def doo(image):
     out = remover.process(img, threshold=0.5) # use threhold parameter for hard prediction.
     
     out.save('output.png') # save result
+    return out
 
 iface = gr.Interface(fn=doo, inputs="image", outputs="image")
 iface.launch()