Ahsen Khaliq commited on
Commit
5db28b0
1 Parent(s): 9504fe5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -166,9 +166,9 @@ def inference(text,steps,image,mode, seed):
166
  return pil_image, "test.mp4"
167
  else:
168
  os.system("python gen_images.py --outdir=out --trunc=1 --seeds="+str(seed)+" \
169
- --network=https://api.ngc.nvidia.com/v2/models/nvidia/research/stylegan3/versions/1/files/stylegan3-r-afhqv2-512x512.pkl")
170
  os.system("python gen_video.py --output=lerp.mp4 --trunc=1 --seeds=0-"+str(seed)+" --grid=1x1 \
171
- --network=https://api.ngc.nvidia.com/v2/models/nvidia/research/stylegan3/versions/1/files/stylegan3-r-afhqv2-512x512.pkl")
172
  img = Image.new("RGB", (800, 1280), (255, 255, 255))
173
  out = Image.open(f'out/seed{seed:04d}.png')
174
  return out, "lerp.mp4"
@@ -180,7 +180,7 @@ article = "<p style='text-align: center'><a href='https://colab.research.google.
180
  examples = [['mario',150,None]]
181
  gr.Interface(
182
  inference,
183
- ["text",gr.inputs.Slider(minimum=50, maximum=200, step=1, default=150, label="steps"),gr.inputs.Image(type="pil", label="Image (Optional)", optional=True),gr.inputs.Radio(["CLIP+StyleGAN3","Stylegan3 interpolation"], type="value", default="CLIP+StyleGAN3", label="mode"),gr.inputs.Slider(minimum=5, maximum=10, step=1, default=5, label="seed (for stylegan3)")],
184
  [gr.outputs.Image(type="pil", label="Output"),"playable_video"],
185
  title=title,
186
  description=description,
 
166
  return pil_image, "test.mp4"
167
  else:
168
  os.system("python gen_images.py --outdir=out --trunc=1 --seeds="+str(seed)+" \
169
+ --network=https://api.ngc.nvidia.com/v2/models/nvidia/research/stylegan3/versions/1/files/stylegan3-r-ffhq-1024x1024.pkl")
170
  os.system("python gen_video.py --output=lerp.mp4 --trunc=1 --seeds=0-"+str(seed)+" --grid=1x1 \
171
+ --network=https://api.ngc.nvidia.com/v2/models/nvidia/research/stylegan3/versions/1/files/stylegan3-r-ffhq-1024x1024.pkl")
172
  img = Image.new("RGB", (800, 1280), (255, 255, 255))
173
  out = Image.open(f'out/seed{seed:04d}.png')
174
  return out, "lerp.mp4"
 
180
  examples = [['mario',150,None]]
181
  gr.Interface(
182
  inference,
183
+ ["text",gr.inputs.Slider(minimum=50, maximum=200, step=1, default=150, label="steps"),gr.inputs.Image(type="pil", label="Image (Optional)", optional=True),gr.inputs.Radio(["CLIP+StyleGAN3","Stylegan3 interpolation"], type="value", default="CLIP+StyleGAN3", label="mode"),gr.inputs.Slider(minimum=5, maximum=100, step=1, default=5, label="seed (for stylegan3)")],
184
  [gr.outputs.Image(type="pil", label="Output"),"playable_video"],
185
  title=title,
186
  description=description,