|
|
from fastapi import FastAPI, HTTPException |
|
|
from pydantic import BaseModel |
|
|
from transformers import pipeline |
|
|
|
|
|
app = FastAPI() |
|
|
|
|
|
classifier = pipeline("text-classification", model="ShazaAly/suplyd-intent-classifier") |
|
|
|
|
|
class IntentRequest(BaseModel): |
|
|
text: str |
|
|
|
|
|
class IntentResponse(BaseModel): |
|
|
label: str |
|
|
confidence: float |
|
|
|
|
|
@app.post("/classify", response_model=IntentResponse) |
|
|
def classify_intent(req: IntentRequest): |
|
|
if not req.text.strip(): |
|
|
return {"label": "غير ذلك", "confidence": 1.0} |
|
|
|
|
|
try: |
|
|
results = classifier(req.text) |
|
|
top = results[0] |
|
|
return { |
|
|
"label": top["label"], |
|
|
"confidence": top["score"] |
|
|
} |
|
|
except Exception as e: |
|
|
raise HTTPException(status_code=500, detail=str(e)) |
|
|
|
