Spaces:

suko
/

nsfw

Running

suko commited on Mar 20, 2023

Commit

2a18958

1 Parent(s): 2ea306a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -95,7 +95,7 @@ def predict(image):
     image = Image.fromarray(np.uint8(image), 'RGB')
     prediction = model.predict(image)
     for output in prediction["predictions"]:
-        output["confidence"] = round(output["confidence"], 2)
     return prediction
 inputs = gr.inputs.Image(type="pil")

     image = Image.fromarray(np.uint8(image), 'RGB')
     prediction = model.predict(image)
     for output in prediction["predictions"]:
+        output["confidence"] = round(output["confidence"], 4)
     return prediction
 inputs = gr.inputs.Image(type="pil")