Update app.py

app.py

@@ -9,6 +9,7 @@ def check(text):
     with torch.no_grad():
         logits = model(**inputs).logits
     predicted_class_id = logits.argmax().item()
+    print(f'{logits.argmax().item()}:{text}')
     return 'secure' if predicted_class_id==0 else 'insecure'
 with gr.Blocks() as demo:
     text = gr.Textbox(label="Text")