Ahsen Khaliq commited on
Commit
ccfbba4
1 Parent(s): 34ad95a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -4,8 +4,9 @@ os.system("gdown https://drive.google.com/uc?id=12ElLliRlgGZqPOhUcqJtNVsa7rmzSI5
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
 
10
  title = "AnimeGANv2"
11
  description = "Gradio Demo for AnimeGanv2 Face Portrait v2. To use it, simply upload your image, or click one of the examples to load them. Read more at the links below. Please use a cropped portrait picture for best results similar to the examples below."
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
8
+ """+style.name+""" --input_img_path """+content.name)
9
+ return "out.jpg"
10
 
11
  title = "AnimeGANv2"
12
  description = "Gradio Demo for AnimeGanv2 Face Portrait v2. To use it, simply upload your image, or click one of the examples to load them. Read more at the links below. Please use a cropped portrait picture for best results similar to the examples below."