Update demo.py

demo.py

@@ -1,17 +1,34 @@
 import gradio as gr
+import spaces
+
+
+# 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 "Prediction: Human (Mocked)"
 
+
+
+# Change mode name
+# def change_mode(mode):
+#   if mode == ""
+
+
+
 css = """
-#header { text-align: center; font-size: 2em; margin-bottom: 15px; }
+#header { text-align: center; font-size: 1.5em; 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 {
@@ -23,7 +40,7 @@ css = """
 # Gradio App
 with gr.Blocks(css=css) as app:
     with gr.Row():
-        gr.HTML('<div id="header">Human or AI Text Detector</div>')
+        gr.HTML('<div id="header">R-detect On HuggingFace</div>')
     with gr.Row():
         gr.HTML(
             """
@@ -39,7 +56,7 @@ with gr.Blocks(css=css) as app:
     with gr.Row():
         input_text = gr.Textbox(
             label="Input Text",
-            placeholder="Enter the text to check",
+            placeholder="Enter Text Here",
             lines=8,
         )
     with gr.Row():
@@ -59,19 +76,20 @@ with gr.Blocks(css=css) as app:
             label="Select Model",
             value="gpt2-medium",
         )
-    with gr.Row():
         submit_button = gr.Button("Run Detection", variant="primary")
         clear_button = gr.Button("Clear", variant="secondary")
     with gr.Row():
         output = gr.Textbox(
-            label="Prediction",
-            placeholder="Prediction: Human or AI",
+            label="Inference Result",
+            placeholder="Maked by Human or AI",
             elem_id="output-text",
         )
     submit_button.click(
         run_test_power, inputs=[model_name, input_text, input_text], outputs=output
     )
     clear_button.click(lambda: ("", ""), inputs=[], outputs=[input_text, output])
+    # model_name.change(change_mode, inputs=[model_name], outputs=[model_name])
+
     with gr.Accordion("Disclaimer", open=False):
         gr.Markdown(
             """