Atlasbot commited on
Commit
ee473f5
·
1 Parent(s): cabe44f

let's deploy to huggingface spaces

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -4,7 +4,8 @@ import skimage
4
 
5
  learn = load_learner('export.pkl')
6
  labels = learn.dls.vocab
7
- def predict(img):
 
8
  img = img.resize((512, 512))
9
  img = PILImage.create(img)
10
  pred,pred_idx,probs = learn.predict(img)
 
4
 
5
  learn = load_learner('export.pkl')
6
  labels = learn.dls.vocab
7
+
8
+ def predict(img):
9
  img = img.resize((512, 512))
10
  img = PILImage.create(img)
11
  pred,pred_idx,probs = learn.predict(img)