Update demo.py

demo.py

@@ -11,13 +11,21 @@ import spaces
 
 # Mock function for testing layout
 def run_test_power(model_name, real_text, generated_text, N=10):
-    return "Prediction: Human (Mocked)"
+    return f"Prediction: Human (Mocked for {model_name})"
 
 
 
 # Change mode name
-# def change_mode(mode):
-#   if mode == ""
+#def change_mode(mode):
+#    if mode == "Faster Model":
+#        .change_mode("t5-small")
+#    elif mode == "Medium Model":
+#        .change_mode("roberta-base-openai-detector")
+#    elif mode == "Powerful Model":
+#        .change_mode("falcon-rw-1b")
+#    else:
+#        gr.Error(f"Invaild mode selected.")
+#    return mode
 
 
 
@@ -59,36 +67,26 @@ with gr.Blocks(css=css) as app:
             placeholder="Enter Text Here",
             lines=8,
         )
+        output = gr.Textbox(
+            label="Inference Result",
+            placeholder="Maked by Human or AI",
+            elem_id="output-text",
+        )
     with gr.Row():
         model_name = gr.Dropdown(
             [
-                "gpt2-medium",
-                "gpt2-large",
-                "t5-large",
-                "t5-small",
-                "roberta-base",
-                "roberta-base-openai-detector",
-                "chatgpt-detector-roberta",
-                "gpt3-small-finetune-cnndaily-news",
-                "gpt-neo-125m",
-                "falcon-rw-1b",
+                "Faster Model",
+                "Medium Model",
+                "Powerful Model",
             ],
             label="Select Model",
             value="gpt2-medium",
         )
         submit_button = gr.Button("Run Detection", variant="primary")
         clear_button = gr.Button("Clear", variant="secondary")
-    with gr.Row():
-        output = gr.Textbox(
-            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
-    )
+    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])
+#    model_name.change(change_mode, inputs=[model_name], outputs=[model_name])
 
     with gr.Accordion("Disclaimer", open=False):
         gr.Markdown(