hysts commited on
Commit
a5bffe4
1 Parent(s): 7a39e7e
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/self-distilled-stylegan/self-distilled-internet-photos'
20
  TITLE = 'Self-Distilled StyleGAN'
21
  DESCRIPTION = f'This is a demo for models provided in {ORIGINAL_REPO_URL}.'
@@ -106,8 +106,6 @@ def load_model(model_name: str, device: torch.device) -> nn.Module:
106
 
107
 
108
  def main():
109
- gr.close_all()
110
-
111
  args = parse_args()
112
  device = torch.device(args.device)
113
 
 
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/self-distilled-stylegan/self-distilled-internet-photos'
20
  TITLE = 'Self-Distilled StyleGAN'
21
  DESCRIPTION = f'This is a demo for models provided in {ORIGINAL_REPO_URL}.'
 
106
 
107
 
108
  def main():
 
 
109
  args = parse_args()
110
  device = torch.device(args.device)
111