Update app.py

app.py

@@ -1,15 +1,25 @@
-from fastapi import FastAPI
+from fastapi import FastAPI, Query
+from pydantic import BaseModel
+from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline
 
 app = FastAPI()
-from transformers import pipeline
 
-pipe = pipeline("text-classification", model="kmcs-casulit/hr_cate")
+# Load the model and tokenizer
+model = AutoModelForSequenceClassification.from_pretrained(
+    "kmcs-casulit/hr_cate")
+tokenizer = AutoTokenizer.from_pretrained("kmcs-casulit/hr_cate")
+pipe = pipeline("text-classification", model=model, tokenizer=tokenizer)
+
+# Define a request model
+class TextRequest(BaseModel):
+    text: str
 
 @app.get("/")
 def greet_json():
     return {"message": "Hello, World!"}
 
-@app.get("/")
-def classify(text: str):
-    output = pipe(text)
+
+@app.post("/classify/")
+def classify(request: TextRequest):
+    output = pipe(request.text)
     return {"output": output[0]['label']}