Spaces:

akhaliq
/

stylegan3_clip

Runtime error

Ahsen Khaliq commited on Oct 22, 2021

Commit

9504fe5

•

1 Parent(s): 2595a70

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -170,7 +170,7 @@ def inference(text,steps,image,mode, seed):
       os.system("python gen_video.py --output=lerp.mp4 --trunc=1 --seeds=0-"+str(seed)+" --grid=1x1 \
     --network=https://api.ngc.nvidia.com/v2/models/nvidia/research/stylegan3/versions/1/files/stylegan3-r-afhqv2-512x512.pkl")
       img = Image.new("RGB", (800, 1280), (255, 255, 255))
-      out = Image.open("out/0002.png")
       return out, "lerp.mp4"

       os.system("python gen_video.py --output=lerp.mp4 --trunc=1 --seeds=0-"+str(seed)+" --grid=1x1 \
     --network=https://api.ngc.nvidia.com/v2/models/nvidia/research/stylegan3/versions/1/files/stylegan3-r-afhqv2-512x512.pkl")
       img = Image.new("RGB", (800, 1280), (255, 255, 255))
+      out = Image.open(f'out/seed{seed:04d}.png')
       return out, "lerp.mp4"