Update demo.py

demo.py

@@ -1,22 +1,21 @@
 import gradio as gr
-import spaces
+from relative_tester import relative_tester
+# from two_sample_tester import two_sample_tester
+from utils import init_random_seeds
 
-
-# TOKENIZER = 
-# MINIMUM_TOKENS = 64
-
-# def count_tokens(text):
-#     return len(TOKENIZER(text).input_ids)
-
-
-# Mock function for testing layout
-def run_test_power(model_name, real_text, generated_text, N=10):
-    return f"Prediction: Human (Mocked for {model_name})"
+init_random_seeds()
 
 
+def run_test(input_text):
+    if not input_text:
+        return "Now that you've built a demo, you'll probably want to share it with others. Gradio demos can be shared in two ways: using a temporary share link or permanent hosting on Spaces."
+    # return two_sample_tester.test(input_text.strip())
+    return relative_tester.test(input_text.strip())
+    return f"Prediction: Human (Mocked for {input_text})"
 
+# TODO: Add model selection in the future
 # Change mode name
-#def change_mode(mode):
+# def change_mode(mode):
 #    if mode == "Faster Model":
 #        .change_mode("t5-small")
 #    elif mode == "Medium Model":
@@ -28,15 +27,14 @@ def run_test_power(model_name, real_text, generated_text, N=10):
 #    return mode
 
 
-
 css = """
 #header { text-align: center; font-size: 3em; margin-bottom: 20px; }
 #output-text { font-weight: bold; font-size: 1.2em; }
-.links { 
-    display: flex; 
-    justify-content: flex-end; 
-    gap: 10px; 
-    margin-right: 10px; 
+.links {
+    display: flex;
+    justify-content: flex-end;
+    gap: 10px;
+    margin-right: 10px;
     align-items: center;
 }
 .separator {
@@ -91,7 +89,8 @@ with gr.Blocks(css=css) as app:
     with gr.Row():
         gr.HTML('<div id="header">R-detect On HuggingFace</div>')
     with gr.Row():
-        gr.HTML("""
+        gr.HTML(
+            """
         <div class="links">
             <a href="https://openreview.net/forum?id=z9j7wctoGV" target="_blank">Paper</a>
             <span class="separator">|</span>
@@ -99,14 +98,14 @@ with gr.Blocks(css=css) as app:
             <span class="separator">|</span>
             <a href="mailto:[email protected]" target="_blank">Contact</a>
         </div>
-        """)
+        """
+        )
     with gr.Row():
         input_text = gr.Textbox(
             label="Input Text",
             placeholder="Enter Text Here",
             lines=8,
             elem_classes=["input-text"],  # Applying the CSS class
-            value="Basketball is a dynamic and widely popular sport that originated in 1891 when Dr. James Naismith, a physical education instructor, sought to keep his students active during the winter. He devised a game involving a ball and two peach baskets, laying the foundation for what would evolve into modern basketball. The objective is to score points by shooting the ball through the opposing team's hoop, a task complicated by the rules and the opposing team's defense. (This is a example from Chatgpt)"
         )
         output = gr.Textbox(
             label="Inference Result",
@@ -116,30 +115,36 @@ with gr.Blocks(css=css) as app:
             elem_classes=["output-text"],
         )
     with gr.Row():
-        model_name = gr.Dropdown(
-            [
-                "Faster Model",
-                "Medium Model",
-                "Powerful Model",
-            ],
-            label="Select Model",
-            value="Medium Model",
-            elem_classes=["select"],
+        # TODO: Add model selection in the future
+        # model_name = gr.Dropdown(
+        #     [
+        #         "Faster Model",
+        #         "Medium Model",
+        #         "Powerful Model",
+        #     ],
+        #     label="Select Model",
+        #     value="Medium Model",
+        #     elem_classes=["select"],
+        # )
+        submit_button = gr.Button(
+            "Run Detection", variant="primary", elem_classes=["button"]
         )
-        submit_button = gr.Button("Run Detection", variant="primary", elem_classes=["button"])
         clear_button = gr.Button("Clear", variant="secondary", elem_classes=["button"])
-    
-    submit_button.click(run_test_power, inputs=[model_name, input_text, input_text], outputs=output)
+
+    submit_button.click(run_test, inputs=[input_text], outputs=output)
     clear_button.click(lambda: ("", ""), inputs=[], outputs=[input_text, output])
 
     with gr.Accordion("Disclaimer", open=False, elem_classes=["accordion"]):
-        gr.Markdown("""
+        gr.Markdown(
+            """
         - **Disclaimer**: This tool is for demonstration purposes only. It is not a foolproof AI detector.
         - **Accuracy**: Results may vary based on input length and quality.
-        """)
+        """
+        )
 
     with gr.Accordion("Citations", open=False, elem_classes=["accordion"]):
-        gr.Markdown("""
+        gr.Markdown(
+            """
         ```
         @inproceedings{zhangs2024MMDMP,
             title={Detecting Machine-Generated Texts by Multi-Population Aware Optimization for Maximum Mean Discrepancy},
@@ -148,6 +153,7 @@ with gr.Blocks(css=css) as app:
             year={2024}
         }
         ```
-        """)
+        """
+        )
 
-app.launch()
+app.launch()