Update app.py

app.py

@@ -1,3 +1,8 @@
+
+
+
+
+
 import gradio as gr
 from detector import CustomDetector
 import logging
@@ -108,41 +113,21 @@ button {
 }
 """
 
-# Additional information for the UI
-info_markdown = """
-# AI-Generated Text Detector
-
-This demo detects AI-generated text using a custom method based on log-likelihood and entropy scoring, powered by the `tiiuae/falcon-rw-1b` model. Enter text to receive a score and classification (AI-generated if score > 0.2, human-generated if score ≤ 0.2).
-
-## Citation
-
-Please cite our work as:
-
-**[Your Paper Title]**  
-[Your Name(s)], [Your Institution/Team]  
-[Paper URL, e.g., https://arxiv.org/abs/your-paper-id]
-
-For details, see the paper linked above. This implementation is for academic use and should be supervised.
-
-*License*: MIT
-"""
-
 # Set up the Gradio interface
-with gr.Blocks(css=custom_css, theme=None) as iface:
-    gr.Markdown(info_markdown)
-    gr.Textbox(
+iface = gr.Interface(
+    fn=detect_text,
+    inputs=gr.Textbox(
         lines=5,
         placeholder="Enter text here to check if it's AI-generated...",
         label="Input Text"
-    )
-    output = gr.Textbox(label="Detection Result")
-    gr.Button("Detect").click(
-        fn=detect_text,
-        inputs=gr.Textbox(),
-        outputs=output
-    )
+    ),
+    outputs=gr.Textbox(label="Detection Result"),
+    title="AI-Generated Text Detector",
+    description="Enter text to detect if it was generated by an AI model. Powered by a custom detector using tiiuae/falcon-rw-1b.",
+    css=custom_css,
+    theme=None
+)
 
 # Launch the app
 if __name__ == "__main__":
-    iface.launch()
-
+    iface.launch()