Update app.py

app.py

@@ -272,19 +272,22 @@ def paraphrase_and_correct(text):
 def detect_ai(text):
     label, score = predict_en(text)
     return label, score
-
-# Gradio interface setup
 def gradio_interface(text):
-    ai_result = detect_ai(text)
+    label, score = detect_ai(text)
     corrected_text = paraphrase_and_correct(text)
-    return ai_result, corrected_text
-
-# Create Gradio interface
-iface = gr.Interface(fn=gradio_interface,
-                     inputs=gr.Textbox(lines=5, placeholder="Enter text here..."),
-                     outputs=[gr.Label(num_top_classes=2), gr.Textbox()],
-                     title="AI Detection and Grammar Correction",
-                     description="Detect AI-generated content and correct grammar issues.")
+    return {label: score}, corrected_text
+
+# Modify the Gradio interface setup
+iface = gr.Interface(
+    fn=gradio_interface,
+    inputs=gr.Textbox(lines=5, placeholder="Enter text here..."),
+    outputs=[
+        gr.Label(num_top_classes=1),
+        gr.Textbox(label="Corrected Text")
+    ],
+    title="AI Detection and Grammar Correction",
+    description="Detect AI-generated content and correct grammar issues."
+)
 
 # Launch the app
 iface.launch()