RaphaelKalandadze commited on
Commit
00d46ee
·
1 Parent(s): b8486cb

run on cpu

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import gradio as gr
2
  from fastai.vision.all import *
3
 
4
- learner = load_learner("models/batch_size_64_device_cuda_lr_0.01_n_epoch_30_seed_42_valid_pct_0.2.pkl", cpu=False)
5
 
6
  labels = learner.dls.vocab
7
 
 
1
  import gradio as gr
2
  from fastai.vision.all import *
3
 
4
+ learner = load_learner("models/batch_size_64_device_cuda_lr_0.01_n_epoch_30_seed_42_valid_pct_0.2.pkl")
5
 
6
  labels = learner.dls.vocab
7