vumichien commited on
Commit
ee94a17
1 Parent(s): 4c7915f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ def get_x(r):
15
  learner = from_pretrained_fastai(repo_id)
16
 
17
  def inference(image):
18
- label_predict,_,probs = learner.predict(img)
19
  return f"This grapevine leave is {label_predict} with {100*probs[torch.argmax(probs)].item():.2f}% probability"
20
 
21
  gr.Interface(
 
15
  learner = from_pretrained_fastai(repo_id)
16
 
17
  def inference(image):
18
+ label_predict,_,probs = learner.predict(image)
19
  return f"This grapevine leave is {label_predict} with {100*probs[torch.argmax(probs)].item():.2f}% probability"
20
 
21
  gr.Interface(