jacksonwambali commited on
Commit
7b995c4
·
verified ·
1 Parent(s): f3d88d2

Create api.py

Browse files
Files changed (1) hide show
  1. api.py +18 -0
api.py ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from fastapi import FastAPI
2
+ from pydantic import BaseModel
3
+ from app import GradioScamDetector
4
+
5
+ app = FastAPI()
6
+ detector = GradioScamDetector()
7
+
8
+ class PredictRequest(BaseModel):
9
+ data: list[str]
10
+
11
+ @app.post("/predict")
12
+ async def predict(request: PredictRequest):
13
+ message = request.data[0]
14
+ result_text, confidence, _, _ = detector.predict_message(message)
15
+ return {
16
+ "result": result_text,
17
+ "confidence": confidence
18
+ }