hysts HF staff commited on
Commit
9f40a8f
1 Parent(s): f9a4051
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
  ORIGINAL_REPO_URL = 'https://github.com/NVlabs/stylegan3'
20
  TITLE = 'NVlabs/stylegan3'
21
  DESCRIPTION = f'This is a demo for {ORIGINAL_REPO_URL}.'
@@ -116,8 +116,6 @@ def load_model(file_name: str, device: torch.device) -> nn.Module:
116
 
117
 
118
  def main():
119
- gr.close_all()
120
-
121
  args = parse_args()
122
  device = torch.device(args.device)
123
 
 
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
  ORIGINAL_REPO_URL = 'https://github.com/NVlabs/stylegan3'
20
  TITLE = 'NVlabs/stylegan3'
21
  DESCRIPTION = f'This is a demo for {ORIGINAL_REPO_URL}.'
 
116
 
117
 
118
  def main():
 
 
119
  args = parse_args()
120
  device = torch.device(args.device)
121