Update app.py

app.py

@@ -4,8 +4,9 @@ os.system("gdown https://drive.google.com/uc?id=12ElLliRlgGZqPOhUcqJtNVsa7rmzSI5
 os.system("gdown https://drive.google.com/uc?id=1-79oBWGFQXrKYw9oxX7t468Zrp87NoWn")
 
 def inference(content, style):
-    os.system("""python style_transfer_folder.py --size 1024 --ckpt ./blendgan.pt --psp_encoder_ckpt ./psp_encoder.pt --style_img_path 
- """+style.name+""" --input_img_path """+content.name)
+    os.rename(content.name, "content.png")
+    os.rename(style.name, "style.png")
+    os.system("""python style_transfer_folder.py --size 1024 --ckpt ./blendgan.pt --psp_encoder_ckpt ./psp_encoder.pt --style_img_path style.png --input_img_path content.png""")
     return "out.jpg"
   
 title = "BlendGAN"