Spaces:
Sleeping
Sleeping
Add entries to requirements.txt and fix model file name
Browse files- app.py +1 -1
- model (1).pkl → cycle-model.pkl +0 -0
- requirements.txt +4 -1
app.py
CHANGED
@@ -5,7 +5,7 @@ import gradio as gr
|
|
5 |
import timm
|
6 |
|
7 |
# Upload your model
|
8 |
-
learn = load_learner('model
|
9 |
|
10 |
categories = learn.dls.vocab
|
11 |
|
|
|
5 |
import timm
|
6 |
|
7 |
# Upload your model
|
8 |
+
learn = load_learner('cycle-model.pkl')
|
9 |
|
10 |
categories = learn.dls.vocab
|
11 |
|
model (1).pkl → cycle-model.pkl
RENAMED
File without changes
|
requirements.txt
CHANGED
@@ -1,2 +1,5 @@
|
|
|
|
1 |
gradio
|
2 |
-
|
|
|
|
|
|
1 |
+
fastai
|
2 |
gradio
|
3 |
+
timm
|
4 |
+
torch
|
5 |
+
torchvision
|