hr16 commited on
Commit
179cbb8
·
1 Parent(s): 0c01bda

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -89,10 +89,11 @@ def inference(seed):
89
  'batch_size': 1,
90
  'pixel_min': -1,
91
  'pixel_max': 1,
92
- }) #https://github.com/adriansahlman/stylegan2_pytorch/blob/master/run_generator.py
 
93
  )[0]
94
 
95
  title = "TWDNEv3 CPU Generator"
96
- description = "Gradio Demo for TWDNEv3 CPU Generator (stylegan2_pytorch port). To use it, simply put your random seed."
97
  article = ""
98
  gr.Interface(inference, ["number"], gr.outputs.Image(type="pil"),title=title,description=description,article=article,allow_flagging=False,allow_screenshot=False).launch()
 
89
  'batch_size': 1,
90
  'pixel_min': -1,
91
  'pixel_max': 1,
92
+ 'gpu': []
93
+ }) #Replace ArgumentParser at run_generator.py
94
  )[0]
95
 
96
  title = "TWDNEv3 CPU Generator"
97
+ description = "Gradio Demo for TWDNEv3 CPU Generator (stylegan2_pytorch port). To use it, simply put your random 32-bit unsigned integer seed."
98
  article = ""
99
  gr.Interface(inference, ["number"], gr.outputs.Image(type="pil"),title=title,description=description,article=article,allow_flagging=False,allow_screenshot=False).launch()