Update app.py

app.py

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