dielz commited on
Commit
823e3a4
·
verified ·
1 Parent(s): 569f7e7

redefine return output

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ def predict_image(image):
21
  top_label = labels[top_index]
22
  top_probability = predictions.numpy()[0][top_index] * 100
23
 
24
- return top_label, top_probability
25
 
26
  # Example images
27
  example_images = [
 
21
  top_label = labels[top_index]
22
  top_probability = predictions.numpy()[0][top_index] * 100
23
 
24
+ return {top_label:top_probability}
25
 
26
  # Example images
27
  example_images = [