Jezia commited on
Commit
5d7d528
·
1 Parent(s): 2d00aa2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,7 +6,7 @@ import wget
6
  import torch
7
  import gradio
8
 
9
- os.system("git clone https://github.com/NVlabs/stylegan3")
10
  sys.path.append('./stylegan3')
11
 
12
  model_names = {
@@ -44,7 +44,7 @@ def load_model(file_name: str, device: torch.device):
44
 
45
  local_path = '/content/'f'{file_name}'
46
  print(local_path)
47
- with open(fetch_model(network_url), 'rb') as f:
48
  model = pickle.load(f)['G_ema']
49
  model.eval()
50
  model.to(device)
 
6
  import torch
7
  import gradio
8
 
9
+ #os.system("git clone https://github.com/NVlabs/stylegan3")
10
  sys.path.append('./stylegan3')
11
 
12
  model_names = {
 
44
 
45
  local_path = '/content/'f'{file_name}'
46
  print(local_path)
47
+ with open(fetch_model(local_path), 'rb') as f:
48
  model = pickle.load(f)['G_ema']
49
  model.eval()
50
  model.to(device)