stylematte

Runtime error

shir0ha commited on Sep 5, 2023

Commit

23fa3d9

1 Parent(s): 81c7e37

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,7 @@ def predict(inp):
     res = inference_img(model, inp)
     print(res.shape, res.dtype)
     main_subject = inp.copy()
-    main_subject = main_subject * (res > 0.1)
     return main_subject
 print("MODEL LOADED")

     res = inference_img(model, inp)
     print(res.shape, res.dtype)
     main_subject = inp.copy()
+    main_subject = main_subject * (res > 0.1)[:,:,np.newaxis]
     return main_subject
 print("MODEL LOADED")