Update app.py

app.py

@@ -19,22 +19,22 @@ def _check_each(text):
     with torch.no_grad():
         logits = _model(**inputs).logits
     predicted_class_id = logits.argmax().item()
-    print(f'{logits.argmax().item()}:{text}')
+    print(f't-{logits.argmax().item()}:{text}')
     return 'secure' if predicted_class_id==0 else 'insecure'
 def _check(text):
     t=text
     while len(t)>512:
-        t=t[0:511]
-        if check_each(t)=='insecure':
+        if check_each(t[0:511])=='insecure':
             return 'insecure'
+        t=t[512:]
     return check_each(t)
 def _check_tiny(text):
     t=text
     while len(t)>512:
-        t=t[0:511]
-        if _check_each(t)=='insecure':
+        if check_each(t[0:511])=='insecure':
             return 'insecure'
-    return _check_each(t)
+        t=t[512:]
+    return check_each(t)
 def check(text):
     return _check(text),_check_tiny(text)
 with gr.Blocks() as demo: