Spaces:

moro23
/

bear-classifier-2

Sleeping

moro23 commited on Apr 29, 2022

Commit

f72b958

1 Parent(s): 4853fee

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ learn_inf = load_learner('bear_classifier_cat')
 def bear_classifier(image):
   pred,pred_idx,probs = learn_inf.predict(image)
   pred_name = 'Prediction: ' + pred
-  pred_prob = 'Probability: ' + probs[pred_idx]:.04f
   return pred_name, pred_prob

 def bear_classifier(image):
   pred,pred_idx,probs = learn_inf.predict(image)
   pred_name = 'Prediction: ' + pred
+  pred_prob = 'Probability: ' + round(probs[pred_idx],4)
   return pred_name, pred_prob