Jezia commited on
Commit
cd015de
1 Parent(s): 5431ad5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -2,6 +2,7 @@ import numpy as np
2
  import pickle as pickle
3
  import os
4
  import sys
 
5
 
6
  #os.system("git clone https://github.com/NVlabs/stylegan3")
7
  #sys.path.append('./stylegan3')
@@ -30,7 +31,7 @@ def fetch_model(url_or_path):
30
  if os.path.exists(basename):
31
  return basename
32
  else:
33
- wget -c f'{url_or_path}'
34
  return basename
35
 
36
  def load_model(file_name: str, device: torch.device):
 
2
  import pickle as pickle
3
  import os
4
  import sys
5
+ import wget
6
 
7
  #os.system("git clone https://github.com/NVlabs/stylegan3")
8
  #sys.path.append('./stylegan3')
 
31
  if os.path.exists(basename):
32
  return basename
33
  else:
34
+ wget.download(url_or_path)
35
  return basename
36
 
37
  def load_model(file_name: str, device: torch.device):