Ahsen Khaliq commited on
Commit
2595a70
1 Parent(s): d1188bd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -165,9 +165,9 @@ def inference(text,steps,image,mode, seed):
165
  writer.close()
166
  return pil_image, "test.mp4"
167
  else:
168
- os.system("python gen_images.py --outdir=out --trunc=1 --seeds="+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-"+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("out/0002.png")
 
165
  writer.close()
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("out/0002.png")