Create app.py

app.py

@@ -0,0 +1,15 @@
+from fastapi import FastAPI
+
+app = FastAPI()
+from transformers import pipeline
+
+pipe = pipeline("text-classification", model="kmcs-casulit/hr_cate")
+
+@app.get("/")
+def greet_json():
+    return {"message": "Hello, World!"}
+
+@app.get("/")
+def classify(text: str):
+    output = pipe(text)
+    return {"output": output[0]['label']}