hysts HF staff commited on
Commit
bb5df6e
1 Parent(s): 62b5d42
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -8,14 +8,14 @@ import os
8
  import pickle
9
  import sys
10
 
11
- sys.path.insert(0, 'stylegan3')
12
-
13
  import gradio as gr
14
  import numpy as np
15
  import torch
16
  import torch.nn as nn
17
  from huggingface_hub import hf_hub_download
18
 
 
 
19
  TITLE = 'StyleGAN2'
20
  DESCRIPTION = '''This is an unofficial demo for https://github.com/NVlabs/stylegan3.
21
 
@@ -135,8 +135,6 @@ def load_model(file_name: str, device: torch.device) -> nn.Module:
135
 
136
 
137
  def main():
138
- gr.close_all()
139
-
140
  args = parse_args()
141
  device = torch.device(args.device)
142
 
 
8
  import pickle
9
  import sys
10
 
 
 
11
  import gradio as gr
12
  import numpy as np
13
  import torch
14
  import torch.nn as nn
15
  from huggingface_hub import hf_hub_download
16
 
17
+ sys.path.insert(0, 'stylegan3')
18
+
19
  TITLE = 'StyleGAN2'
20
  DESCRIPTION = '''This is an unofficial demo for https://github.com/NVlabs/stylegan3.
21
 
 
135
 
136
 
137
  def main():
 
 
138
  args = parse_args()
139
  device = torch.device(args.device)
140