Spaces:

MsChabane
/

RustDetective

Sleeping

MsChabane commited on 17 days ago

Commit

5920752

verified ·

1 Parent(s): ea32d4d

Update app/routes.py

Files changed (1) hide show

app/routes.py CHANGED Viewed

@@ -39,7 +39,7 @@ def upload(image:UploadFile=File(...)):
             uri =detect(image_bytes)
         response ={
             "prediction":CLASSES[class_number],
-            "probability":round(probability,4)*100,
             "uri":uri
         }
         return JSONResponse(content=response,status_code=200)

             uri =detect(image_bytes)
         response ={
             "prediction":CLASSES[class_number],
+            "probability":str(round(probability,4)*100),
             "uri":uri
         }
         return JSONResponse(content=response,status_code=200)