Initial commit

app.py

@@ -0,0 +1,14 @@
+from fastapi import FastAPI
+from transformers import pipeline
+from pydantic import BaseModel
+
+app = FastAPI()
+detector = pipeline("text-classification", model="roberta-base-openai-detector")
+
+class TextInput(BaseModel):
+    text: str
+
+@app.post("/predict")
+def predict(input: TextInput):
+    result = detector(input.text)
+    return {"prediction": result[0]['label'], "score": result[0]['score']}

requirements.txt

@@ -0,0 +1,5 @@
+transformers
+torch
+fastapi
+uvicorn
+pydantic