Update app.py

app.py

@@ -44,7 +44,25 @@ def predict_vihatet5(comment: str, prefix: str = "hate-speech-detection"):
     output_ids = model.generate(**inputs, max_length=50)
     output = tokenizer.decode(output_ids[0], skip_special_tokens=True)
     return output
-
+@app.get("/")
+async def root():
+    """Health check endpoint with comprehensive API documentation"""
+    return {
+        "endpoints": {
+            "POSR /predict": {
+                "Resquest body": {
+                    "input": "string",
+                    "prefix": "hate-speech-detection or toxic-speech-detection"
+                },
+                "Response": {
+                    "input": "string",
+                    "prefix": "hate-speech-detection or toxic-speech-detection",
+                    "prediction": "offensive,hate,..."
+                    
+                }
+            }
+        }
+    }
 # API route
 @app.post("/predict")
 def predict(input_data: CommentInput):