Add logs

app.py

@@ -70,6 +70,8 @@ def get_robot_txt():
 def predict(model: InputModel) -> OutputModel:
     text = model.query.message.text
 
+    print(f"Predict: {text}")
+
     urls = extract_urls(text)
     results = [urlscan.scan(url) for url in urls]
 
@@ -83,9 +85,11 @@ def predict(model: InputModel) -> OutputModel:
 
     label = pipe(text)
     if label[0]['label'] == 'LABEL_1':
+        print("Classify LABEL_1. Submitting entry and returning JUNK.")
         submit_entry(model.query.sender, model.query.message.text)
         return OutputModel(action=ActionModel.JUNK, sub_action=SubActionModel.NONE)
     else:
+        print("Classify LABEL_0. Submitting entry and returning NONE.")
         return OutputModel(action=ActionModel.NONE, sub_action=SubActionModel.NONE)
 
 class ReportModel(BaseModel):