ali-ghamdan commited on
Commit
ac5fd2f
Β·
1 Parent(s): c11bb92

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -1,14 +1,15 @@
1
  import os
 
 
 
2
  import gradio as gr
3
  from deoldify import device
4
  from deoldify.device_id import DeviceId
5
  from deoldify.visualize import *
6
  import tempfile
7
 
8
- if not os.path.exists("models/ColorizeArtistic_gen.pth"):
9
- os.system(
10
- 'wget "https://data.deepai.org/deoldify/ColorizeArtistic_gen.pth" -O "models/ColorizeArtistic_gen.pth"'
11
- )
12
 
13
  device.set(device=DeviceId.GPU0)
14
 
 
1
  import os
2
+
3
+ os.system('wget "https://data.deepai.org/deoldify/ColorizeArtistic_gen.pth" -O "models/ColorizeArtistic_gen.pth"')
4
+
5
  import gradio as gr
6
  from deoldify import device
7
  from deoldify.device_id import DeviceId
8
  from deoldify.visualize import *
9
  import tempfile
10
 
11
+ # if not os.path.exists("models/ColorizeArtistic_gen.pth"):
12
+
 
 
13
 
14
  device.set(device=DeviceId.GPU0)
15