Update app.py

app.py

@@ -14,9 +14,9 @@ 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]
+    print(res.shape, res.type)
+    # main_subject = inp.copy()
+    # main_subject = main_subject * res[res > 0.1]
     return main_subject
 
 print("MODEL LOADED")