yuragoithf commited on
Commit
a0c91d6
·
verified ·
1 Parent(s): 109e44b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -59,7 +59,7 @@ def predict_class(image):
59
 
60
  # UI Design for multy class output
61
  def classify_image(image):
62
- results = predict_class(image)
63
  print("results is ...", results)
64
  output = {labels.get(i): float(results[i]) for i in range(len(results))}
65
  print("output is ...", output)
 
59
 
60
  # UI Design for multy class output
61
  def classify_image(image):
62
+ results = predict_class(image)[0]
63
  print("results is ...", results)
64
  output = {labels.get(i): float(results[i]) for i in range(len(results))}
65
  print("output is ...", output)