Spaces:

satani
/

bird_classifier

Runtime error

satani commited on Jul 24, 2022

Commit

64851c7

1 Parent(s): e1946a3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,8 +1,7 @@
 from fastai.vision.all import *
 import gradio as gr
-path = Path()
-learn_inf = load_learner(path/'export.pkl')
 def classify_image(img):
     pred,pred_idx,probs = learn_inf.predict(img)
     return f'Prediction: {pred}; Probability: float({probs[pred_idx]:.04f})'

 from fastai.vision.all import *
 import gradio as gr
+learn_inf = load_learner('export.pkl')
 def classify_image(img):
     pred,pred_idx,probs = learn_inf.predict(img)
     return f'Prediction: {pred}; Probability: float({probs[pred_idx]:.04f})'