sagivp commited on
Commit
8aeb811
·
1 Parent(s): cad6602

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -11
app.py CHANGED
@@ -1,15 +1,6 @@
1
- import os
2
- os.system('pip uninstall torch torchvision torchaudio')
3
-
4
- os.system('pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cpu')
5
-
6
-
7
- ###
8
  import gradio as gr
9
  from fastai.vision.all import *
10
 
11
- defaults.device = 'cpu'
12
- torch.cuda.set_device('cpu')
13
  learn = load_learner('model.pkl')
14
 
15
 
@@ -34,5 +25,4 @@ gr.Interface(fn=predict,
34
  article=article,
35
  examples=examples,
36
  interpretation=interpretation,
37
- enable_queue=enable_queue).launch()
38
- ###
 
 
 
 
 
 
 
 
1
  import gradio as gr
2
  from fastai.vision.all import *
3
 
 
 
4
  learn = load_learner('model.pkl')
5
 
6
 
 
25
  article=article,
26
  examples=examples,
27
  interpretation=interpretation,
28
+ enable_queue=enable_queue).launch()