update predict size

src/online_prediction.py

@@ -20,6 +20,8 @@ def consume_kafka(device_dict, diary_dict, tractate_dict, diary_predict_fn, trac
     consumer.poll()
     consumer.seek_to_end()
 
+    predict_size = 300
+
     try:
         for msg in consumer:
             data = json.loads(msg.value)
@@ -28,9 +30,9 @@ def consume_kafka(device_dict, diary_dict, tractate_dict, diary_predict_fn, trac
             event_cn = data.get("event_cn", "")
             # if device_id == "androidid_a25a1129c0b38f7b":
             print(device_id + " " + log_time + " " + event_cn + " " + str(datetime.now()))
-            offline_predict_diary(device_id, device_dict, diary_dict, diary_predict_fn, size=300)
-            print("---------------------")
-            offline_predict_tractate(device_id, device_dict, tractate_dict, tractate_predict_fn, size=300)
+            offline_predict_diary(device_id, device_dict, diary_dict, diary_predict_fn, size=predict_size)
+            print("-------------------")
+            offline_predict_tractate(device_id, device_dict, tractate_dict, tractate_predict_fn, size=predict_size)
             print("=========================================\n")
     except Exception as e:
         print(e)