Update app.py

app.py

@@ -1,10 +1,14 @@
-import gradio as gr
+from fastapi import FastAPI
+from pydantic import BaseModel
 from transformers import pipeline
 
+app = FastAPI()
 moderate_pipe = pipeline("text-classification", model="KoalaAI/Text-Moderation")
 
-def moderate(text):
-    results = moderate_pipe(text)
-    return {r["label"]: r["score"] for r in results}
+class TextInput(BaseModel):
+    text: str
 
-gr.Interface(fn=moderate, inputs="text", outputs="label").launch()
+@app.post("/moderate")
+async def moderate_text(input: TextInput):
+    results = moderate_pipe(input.text)
+    return {r["label"]: r["score"] for r in results}