Update app.py

app.py

@@ -1,19 +1,13 @@
 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']
+from PIL import Image
 
 os.system("wget https://www.dropbox.com/s/fgupbov77x4rrru/blendgan.pt")
 os.system("wget https://www.dropbox.com/s/v8q0dd3r4u20659/psp_encoder.pt")
-import matplotlib.pyplot as plt
 
 def inference(content, style):
-    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')
+    content.save('content.png')
+    style.save('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"