Hugging Face's logo

Spaces:
ludjan
/
space-bike
Sleeping

App Files Community
ludjan commited on
Commit
b8a8411
•
1 Parent(s): 66bac3d

Add entries to requirements.txt and fix model file name

Browse files
Files changed (3) hide show
  1. app.py +1 -1
  2. model (1).pkl → cycle-model.pkl +0 -0
  3. 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 (1).pkl')
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
- fastai
 
 
 
1
+ fastai
2
  gradio
3
+ timm
4
+ torch
5
+ torchvision