Update app.py

app.py

@@ -165,10 +165,13 @@ def inference(text,steps,image,mode):
       writer.close()
       return pil_image, "test.mp4"
   else:
-      os.system("python gen_video.py --output=lerp.mp4 --trunc=1 --seeds=0-31 --grid=4x2 \
+      os.system("python gen_images.py --outdir=out --trunc=1 --seeds=2 \
+    --network=https://api.ngc.nvidia.com/v2/models/nvidia/research/stylegan3/versions/1/files/stylegan3-r-afhqv2-512x512.pkl")
+      os.system("python gen_video.py --output=lerp.mp4 --trunc=1 --seeds=0-10 --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))
-      return img, "lerp.mp4"
+      out = Image.open("out/2.png")
+      return out, "lerp.mp4"
   
   
 title = "StyleGAN3+CLIP"