Spaces:
Sleeping
Sleeping
File size: 559 Bytes
ddcec3b 6cbee4e ddcec3b 6cbee4e ddcec3b fd849c0 6cbee4e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 | from fastapi import FastAPI
from .schemas import PredictRequest, PredictResponse
from .detector import predict_spam
app = FastAPI(
title="Semantic SMS Spam Detection API",
description="Embedding-based spam detection using sentence transformers",
version="1.0.0"
)
@app.get("/")
def status():
return {
"status": "ok",
"model": "all-MiniLM-L6-v2",
"method": "semantic similarity"
}
@app.post("/predict", response_model=PredictResponse)
def predict(request: PredictRequest):
return predict_spam(request.text)
|