Add simpler API endpoint

app.py

@@ -25,17 +25,19 @@ def detect_ai_text(text):
     ai_prob = probs[0][1].item()
     
     return {
-        "Human": f"{human_prob * 100:.2f}%",
-        "AI Generated": f"{ai_prob * 100:.2f}%"
+        "human_probability": round(human_prob * 100, 2),
+        "ai_probability": round(ai_prob * 100, 2),
+        "prediction": "AI Generated" if ai_prob > 0.5 else "Human Written"
     }
 
-# Create Gradio interface
+# Create Gradio interface with API endpoint
 iface = gr.Interface(
     fn=detect_ai_text,
     inputs=gr.Textbox(lines=10, placeholder="Enter text to analyze..."),
     outputs=gr.JSON(label="Detection Results"),
     title="AI Text Detector",
-    description="Detects whether text was written by a human or generated by AI"
+    description="Detects whether text was written by a human or generated by AI",
+    api_name="detect"
 )
 
 if __name__ == "__main__":