moro23 commited on
Commit
03307da
·
1 Parent(s): c6c9a02

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,8 +5,8 @@ learn_inf = load_learner('bear_classifier_cat')
5
 
6
  def bear_classifier(image):
7
  pred,pred_idx,probs = learn_inf.predict(image)
8
- pred_name = f"Prediction: , {pred}"
9
- pred_prob = f"Probability: , {probs[pred_idx]}"
10
  return pred_name, pred_prob
11
 
12
 
 
5
 
6
  def bear_classifier(image):
7
  pred,pred_idx,probs = learn_inf.predict(image)
8
+ pred_name = f"Prediction: {pred}"
9
+ pred_prob = f"Probability: {probs[pred_idx]}"
10
  return pred_name, pred_prob
11
 
12