ali-ghamdan commited on
Commit
c11bb92
Β·
1 Parent(s): 3879744
Files changed (2) hide show
  1. app.py +2 -2
  2. models/.keep +0 -0
app.py CHANGED
@@ -5,9 +5,9 @@ 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)
 
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)
models/.keep ADDED
File without changes