Update app.py

app.py

@@ -1,13 +1,19 @@
 import os 
 import gradio as gr
+os.system('pip install matplotlib')
+os.system('pip install deepface')
+from deepface import DeepFace
+backends = ['opencv', 'ssd', 'dlib', 'mtcnn']
+
 os.system("wget https://www.dropbox.com/s/fgupbov77x4rrru/blendgan.pt")
 os.system("wget https://www.dropbox.com/s/v8q0dd3r4u20659/psp_encoder.pt")
-
-from PIL import Image
+import matplotlib.pyplot as plt
 
 def inference(content, style):
-    content.save('content.png')
-    style.save('style.png')
+    conimg = DeepFace.detectFace(content.name, detector_backend = backends[0])
+    plt.imsave('content.png', conimg, cmap='Greys')
+    styleimg = DeepFace.detectFace(style.name, detector_backend = backends[0])
+    plt.imsave('style.png', styleimg, cmap='Greys')
     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"