Update app.py

app.py

@@ -15,9 +15,10 @@ model.eval()
 def predict(inp):
     print("***********Inference****************")
     mask = inference_img(model, inp) 
-    print("***********Inference finish****************")
     inp_np = np.array(inp)
     fg = np.uint8((mask*inp_np).permute(1,2,0).numpy())
+    print("***********Inference finish****************")
+    print("***********FG****************", fg.shape, mask.shape)
     
     return [mask, fg]