Update app.py

app.py

@@ -1,7 +1,6 @@
 from fastapi import FastAPI, Request
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
-import uvicorn
 
 app = FastAPI(title="AI Detector API")
 
@@ -11,7 +10,6 @@ tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 model = AutoModelForSequenceClassification.from_pretrained(MODEL_NAME)
 model.eval()
 
-
 def get_ai_probability(text: str) -> float:
     """Return the AI probability (0–100%) for a given text."""
     inputs = tokenizer(text, return_tensors="pt", truncation=True, max_length=512)
@@ -21,15 +19,15 @@ def get_ai_probability(text: str) -> float:
     ai_score = probs[0][1].item() * 100
     return round(ai_score, 2)
 
+@app.get("/")
+async def home():
+    return {
+        "message": "Welcome to Saber’s AI Detector API 👋",
+        "usage": "Send a POST request to /analyze with JSON {'text': 'your article text'}"
+    }
 
 @app.post("/analyze")
 async def analyze_text(request: Request):
-    """
-    Example body:
-    {
-      "text": "Your long article text here"
-    }
-    """
     data = await request.json()
     text = data.get("text", "").strip()
     if not text:
@@ -53,7 +51,3 @@ async def analyze_text(request: Request):
         "overall_human_score": round(100 - overall, 2),
         "paragraphs": results
     }
-
-
-if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=7860)