DanielViniciusAlves commited on
Commit
cd16ed8
1 Parent(s): c60c6d9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,6 +1,7 @@
1
- from fastai import load_learner
2
 
3
  learn = load_learner('model.pkl')
 
4
  labels = learn.dls.vocab
5
  def predict(img):
6
  img = PILImage.create(img)
 
1
+ from fastai.vision.all import *
2
 
3
  learn = load_learner('model.pkl')
4
+
5
  labels = learn.dls.vocab
6
  def predict(img):
7
  img = PILImage.create(img)