Spaces:

zhongdj
/

minimal

Sleeping

zhongdj commited on Mar 31, 2023

Commit

71a2079

1 Parent(s): 992fc6d

fix app

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,7 @@ f = Path() / 'bamboo_or_butterfly.pkl'
 learn_inf = load_learner(f)
 def predict(img):
-    pred, pred_idx, probs = learn_inf.predict(img.to_thumb(224))
     return f'Prediction: {pred}; Probability:{probs[pred_idx]:.04f}'
 image = gr.inputs.Image(shape=(192, 192))

 learn_inf = load_learner(f)
 def predict(img):
+    pred, pred_idx, probs = learn_inf.predict(img)
     return f'Prediction: {pred}; Probability:{probs[pred_idx]:.04f}'
 image = gr.inputs.Image(shape=(192, 192))