罗俊辉 hace 1 año
padre
commit
5a803d7fe2
Se han modificado 1 ficheros con 5 adiciones y 5 borrados
  1. 5 5
      applications/functions.py

+ 5 - 5
applications/functions.py

@@ -57,13 +57,13 @@ class ParamProcess(object):
                     obj = {
                         "score": result[0],
                         "benchmark": 0.06,
-                        "is_good_video": result[0] > 0.06
+                        "is_good_video": 1 if result[0] > 0.06 else 0
                     }
                 else:
                     obj = {
                         "score": None,
                         "benchmark": 0.06,
-                        "is_good_video": False
+                        "is_good_video": 0
                     }
                 return obj
             case "v2":
@@ -73,13 +73,13 @@ class ParamProcess(object):
                     obj = {
                         "score": result[0],
                         "benchmark": 0.3,
-                        "is_good_video": result[0] > 0.3
+                        "is_good_video": 1 if result[0] > 0.3 else 0
                     }
                 else:
                     obj = {
                         "score": None,
                         "benchmark": 0.3,
-                        "is_good_video": False
+                        "is_good_video": 0
                     }
                 return obj
 
@@ -121,5 +121,5 @@ class ParamProcess(object):
         :return:
         """
         version, features = await self.process_label(params)
-        print(version, features)
+        # print(version, features)
         return await self.predict_score(version, features)