Update app.py

app.py

@@ -1,7 +1,8 @@
 import gradio as gr
 from transformers import pipeline
 
-clf = pipeline("text-classification", model="MayZhou/e5-small-lora-ai-generated-detector")
+# Use a stable, supported model
+clf = pipeline("text-classification", model="roberta-base-openai-detector")
 
 def detect_ai(text):
     if not text.strip():
@@ -14,12 +15,12 @@ def detect_ai(text):
     except Exception as e:
         return f"Error: {str(e)}"
 
-# Launch the app with a basic Gradio Interface (exposes /predict endpoint)
+# Simple Interface to expose API at /api/predict/
 demo = gr.Interface(
     fn=detect_ai,
     inputs=gr.Textbox(lines=4, label="Enter text"),
     outputs="text",
-    title="AI Text Detector",
+    title="AI Text Detector"
 )
 
 demo.launch()