Update demo/demo.py

demo/demo.py

@@ -1,4 +1,27 @@
+__all__ = ["app"]
 
+import gradio as gr
+import spaces
+from binoculars import Binoculars
+
+BINO = Binoculars(quantize=True)
+TOKENIZER = BINO.tokenizer
+MINIMUM_TOKENS = 64
+
+
+def count_tokens(text):
+    return len(TOKENIZER(text).input_ids)
+
+
+@spaces.GPU
+def run_detector(input_str):
+    if count_tokens(input_str) < MINIMUM_TOKENS:
+        gr.Warning(f"Too short length. Need minimum {MINIMUM_TOKENS} tokens to run Binoculars.")
+        return ""
+    return f"{BINO.predict(input_str)}"
+
+
+def change_mode(mode):
     if mode == "Low False Positive Rate":
         BINO.change_mode("low-fpr")
     elif mode == "High Accuracy":
@@ -15,14 +38,25 @@
 #     return ["Loaded"] * 2
 
 
+css = """
+.green { color: black!important;line-height:1.9em; padding: 0.2em 0.2em; background: #ccffcc; border-radius:0.5rem;}
+.red { color: black!important;line-height:1.9em; padding: 0.2em 0.2em; background: #ffad99; border-radius:0.5rem;}
+.hyperlinks {
+  display: flex;
+  align-items: center;
+  align-content: center;
+  padding-top: 12px;
+  justify-content: flex-end;
+  margin: 0 10px; /* Adjust the margin as needed */
+  text-decoration: none;
+  color: #000; /* Set the desired text color */
+}
+"""
+
 # Most likely human generated, #most likely AI written
 
 capybara_problem = '''Dr. Capy Cosmos, a capybara unlike any other, astounded the scientific community with his groundbreaking research in astrophysics. With his keen sense of observation and unparalleled ability to interpret cosmic data, he uncovered new insights into the mysteries of black holes and the origins of the universe. As he peered through telescopes with his large, round eyes, fellow researchers often remarked that it seemed as if the stars themselves whispered their secrets directly to him. Dr. Cosmos not only became a beacon of inspiration to aspiring scientists but also proved that intellect and innovation can be found in the most unexpected of creatures.'''
 
-tailwind_cdn = """
-<script src="https://cdn.tailwindcss.com"></script>
-"""
-
 with gr.Blocks(css=css,
                theme=gr.themes.Default(font=[gr.themes.GoogleFont("Inconsolata"), "Arial", "sans-serif"])) as app:
     with gr.Row():
@@ -97,4 +131,4 @@ with gr.Blocks(css=css,
     # clear_button.click(lambda x: input_box., )
     submit_button.click(run_detector, inputs=input_box, outputs=output_text)
     clear_button.click(lambda: ("", ""), outputs=[input_box, output_text])
-    dropdown_mode.change(change_mode, inputs=[dropdown_mode], outputs=[dropdown_mode])
+    dropdown_mode.change(change_mode, inputs=[dropdown_mode], outputs=[dropdown_mode])