sparrow8i8 commited on
Commit
05277c0
·
1 Parent(s): e1a402d

added timm requirement

Browse files
Files changed (2) hide show
  1. app.py +1 -0
  2. requirements.txt +2 -1
app.py CHANGED
@@ -1,6 +1,7 @@
1
  import gradio as gr
2
  import fastai
3
  import skimage
 
4
  from fastai.vision.all import *
5
 
6
  learn=load_learner('model.pkl')
 
1
  import gradio as gr
2
  import fastai
3
  import skimage
4
+ import timm
5
  from fastai.vision.all import *
6
 
7
  learn=load_learner('model.pkl')
requirements.txt CHANGED
@@ -16,4 +16,5 @@ session_info==1.0.0
16
  scikit-learn==1.2.2
17
  torch==2.0.0
18
  torchvision==0.15.1
19
- scikit-image
 
 
16
  scikit-learn==1.2.2
17
  torch==2.0.0
18
  torchvision==0.15.1
19
+ scikit-image
20
+ timm