Ahsen Khaliq commited on
Commit
1d11e4b
1 Parent(s): 455bb17

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ import gradio as gr
3
  os.system("gdown https://drive.google.com/uc?id=12ElLliRlgGZqPOhUcqJtNVsa7rmzSI5L")
4
  os.system("gdown https://drive.google.com/uc?id=1-79oBWGFQXrKYw9oxX7t468Zrp87NoWn")
5
 
6
- def inference(img, ver):
7
  os.system("""python style_transfer_folder.py --size 1024 --ckpt ./blendgan.pt --psp_encoder_ckpt ./psp_encoder.pt --style_img_path /content/BlendGAN/style/ --input_img_path /content/BlendGAN/input/ --outdir results/style_transfer/""")
8
  return out
9
 
 
3
  os.system("gdown https://drive.google.com/uc?id=12ElLliRlgGZqPOhUcqJtNVsa7rmzSI5L")
4
  os.system("gdown https://drive.google.com/uc?id=1-79oBWGFQXrKYw9oxX7t468Zrp87NoWn")
5
 
6
+ def inference(content, style):
7
  os.system("""python style_transfer_folder.py --size 1024 --ckpt ./blendgan.pt --psp_encoder_ckpt ./psp_encoder.pt --style_img_path /content/BlendGAN/style/ --input_img_path /content/BlendGAN/input/ --outdir results/style_transfer/""")
8
  return out
9