Update app.py

app.py

@@ -104,7 +104,7 @@ zs = torch.randn([10000, G.mapping.z_dim], device=device)
 w_stds = G.mapping(zs, None).std(0)
     
     
-def inference(text,steps,image,mode):
+def inference(text,steps,image,mode, seed):
   if mode == "CLIP+StyleGAN3":
       all_frames = []
       target = clip_model.embed_text(text)
@@ -165,12 +165,12 @@ def inference(text,steps,image,mode):
       writer.close()
       return pil_image, "test.mp4"
   else:
-      os.system("python gen_images.py --outdir=out --trunc=1 --seeds=2 \
+      os.system("python gen_images.py --outdir=out --trunc=1 --seeds="+seed+" \
     --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 \
+      os.system("python gen_video.py --output=lerp.mp4 --trunc=1 --seeds=0-"+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/2.png")
+      out = Image.open("out/0002.png")
       return out, "lerp.mp4"
   
   
@@ -180,7 +180,7 @@ article = "<p style='text-align: center'><a href='https://colab.research.google.
 examples = [['mario',150,None]]
 gr.Interface(
     inference, 
-    ["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")], 
+    ["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)")], 
     [gr.outputs.Image(type="pil", label="Output"),"playable_video"],
     title=title,
     description=description,