sqlbipro commited on
Commit
eb04f86
·
1 Parent(s): e30e94b
Files changed (2) hide show
  1. app.py +1 -0
  2. requirements.txt +2 -1
app.py CHANGED
@@ -5,6 +5,7 @@ import skimage
5
  learn = load_learner('export.pkl')
6
 
7
  labels = learn.dls.vocab
 
8
  def predict(img):
9
  img = PILImage.create(img)
10
  pred,pred_idx,probs = learn.predict(img)
 
5
  learn = load_learner('export.pkl')
6
 
7
  labels = learn.dls.vocab
8
+
9
  def predict(img):
10
  img = PILImage.create(img)
11
  pred,pred_idx,probs = learn.predict(img)
requirements.txt CHANGED
@@ -1,2 +1,3 @@
1
  fastai==2.7.19
2
- scikit-image
 
 
1
  fastai==2.7.19
2
+ scikit-image
3
+ gradio