Update app.py

app.py

@@ -17,7 +17,7 @@ def predict(inp):
     print(res.shape, res.dtype)
     # main_subject = inp.copy()
     # main_subject = main_subject * res[res > 0.1]
-    return main_subject
+    return res
 
 print("MODEL LOADED")
 print("************************************")