Spaces:

moro23
/

bear-classifier-2

Sleeping

moro23 commited on May 2, 2022

Commit

d3e5077

1 Parent(s): 03307da

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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

 def bear_classifier(image):
   pred,pred_idx,probs = learn_inf.predict(image)
   pred_name = f"Prediction:  {pred}"
+  pred_prob = f"Probability:  {probs[pred_idx]:.04f}"
   return pred_name, pred_prob