yuragoithf commited on
Commit
334672e
·
1 Parent(s): d611617

Update app.py

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