初始化

applications/functions.py

@@ -38,7 +38,7 @@ class ParamProcess(object):
                 features['tag2'] = None
                 features['tag3'] = None
         df = pd.DataFrame([features])
-        print("data_frame", df.columns)
+        # print("data_frame", df.columns)
         df = df.drop('title', axis=1)
         return df
 
@@ -55,12 +55,14 @@ class ParamProcess(object):
                 result = list(result)
                 if result:
                     obj = {
+                        "vision": "v1",
                         "score": result[0],
                         "benchmark": 0.06,
                         "is_good_video": 1 if result[0] > 0.06 else 0
                     }
                 else:
                     obj = {
+                        "vision": "v1",
                         "score": None,
                         "benchmark": 0.06,
                         "is_good_video": 0
@@ -71,12 +73,14 @@ class ParamProcess(object):
                 result = list(result)
                 if result:
                     obj = {
+                        "vision": "v2",
                         "score": result[0],
                         "benchmark": 0.3,
                         "is_good_video": 1 if result[0] > 0.3 else 0
                     }
                 else:
                     obj = {
+                        "vision": "v2",
                         "score": None,
                         "benchmark": 0.3,
                         "is_good_video": 0