Update app.py

app.py

@@ -593,16 +593,22 @@ def build_outputs(results, descriptions, result_details, plate_number):
 
 app = FastAPI()
 
+class InputData(BaseModel):
+    message: str
+    threshold: float = 0.0005
+    tarif_pkb: float = 0.015
+    tarif_bbnkb: float = 0.1
+
 @app.post("/pred")
-async def greet_json(
-    message : str, 
-    threshold : float = 0.0005,
-    tarif_pkb : float = 0.015,
-    tarif_bbnkb : float = 0.1,
-):
+async def greet_json(body: InputData):
     global codes_emb
     global undetected
-
+    
+    message = body.message
+    threshold = body.threshold
+    tarif_pkb = body.tarif_pkb
+    tarif_bbnkb = body.tarif_bbnkb
+    
     undetected_code = undetected[:3]
     
     # tarif_pkb = 1.5/100