Update app.py

app.py

@@ -14,6 +14,7 @@ model.eval()
 def predict(inp):
     print("***********Inference****************")
     res = inference_img(model, inp) 
+    print(res.shape)
     main_subject = inp.copy()
     main_subject = main_subject * res[res > 0.1]
     return main_subject