Update app.py
Browse files
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()
|
|