hysts HF staff commited on
Commit
99a0547
1 Parent(s): 3f3b93c
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, 'stylegan_xl')
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/autonomousvision/stylegan_xl'
20
  TITLE = 'autonomousvision/stylegan_xl'
21
  DESCRIPTION = f'''This is a demo for {ORIGINAL_REPO_URL}.
@@ -124,8 +124,6 @@ def load_model(model_name: str, device: torch.device) -> nn.Module:
124
 
125
 
126
  def main():
127
- gr.close_all()
128
-
129
  args = parse_args()
130
  device = torch.device(args.device)
131
 
 
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, 'stylegan_xl')
18
+
19
  ORIGINAL_REPO_URL = 'https://github.com/autonomousvision/stylegan_xl'
20
  TITLE = 'autonomousvision/stylegan_xl'
21
  DESCRIPTION = f'''This is a demo for {ORIGINAL_REPO_URL}.
 
124
 
125
 
126
  def main():
 
 
127
  args = parse_args()
128
  device = torch.device(args.device)
129