suko commited on
Commit
2a18958
1 Parent(s): 2ea306a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -95,7 +95,7 @@ def predict(image):
95
  image = Image.fromarray(np.uint8(image), 'RGB')
96
  prediction = model.predict(image)
97
  for output in prediction["predictions"]:
98
- output["confidence"] = round(output["confidence"], 2)
99
  return prediction
100
 
101
  inputs = gr.inputs.Image(type="pil")
 
95
  image = Image.fromarray(np.uint8(image), 'RGB')
96
  prediction = model.predict(image)
97
  for output in prediction["predictions"]:
98
+ output["confidence"] = round(output["confidence"], 4)
99
  return prediction
100
 
101
  inputs = gr.inputs.Image(type="pil")