Update app.py

app.py

@@ -1,14 +1,15 @@
 import os
 import onnxruntime as rt
 import sys
-import PIL
-from PIL import Image, ImageOps, ImageFile
+from PIL import Image, ImageOps
 import numpy as np
 from pathlib import Path
 import collections
 from typing import Union, List
 import scipy.ndimage
 import requests
+import gradio as gr
+import face_detection  # Ensure this is the adjusted face_detection.py
 
 MODEL_FILE = "ffhqu2vintage512_pix2pixHD_v1E11-inp2inst-simp.onnx"
 so = rt.SessionOptions()
@@ -20,27 +21,25 @@ print("input_name = " + str(input_name))
 output_name = session.get_outputs()[0].name
 print("output_name = " + str(output_name))
 
-import face_detection
-
 def array_to_image(array_in):
-    array_in = np.squeeze(255*(array_in + 1)/2)
+    array_in = np.squeeze(255 * (array_in + 1) / 2)
     array_in = np.transpose(array_in, (1, 2, 0))
     im = Image.fromarray(array_in.astype(np.uint8))
     return im
 
 def image_as_array(image_in):
     im_array = np.array(image_in, np.float32)
-    im_array = (im_array/255)*2 - 1
+    im_array = (im_array / 255) * 2 - 1
     im_array = np.transpose(im_array, (2, 0, 1))
     im_array = np.expand_dims(im_array, 0)
     return im_array
 
-def find_aligned_face(image_in, size=512):   
+def find_aligned_face(image_in, size=512):
     aligned_image, n_faces, quad = face_detection.align(image_in, face_index=0, output_size=size)
     return aligned_image, n_faces, quad
 
 def align_first_face(image_in, size=512):
-    aligned_image, n_faces, quad = find_aligned_face(image_in,size=size)
+    aligned_image, n_faces, quad = find_aligned_face(image_in, size=size)
     if n_faces == 0:
         try:
             image_in = ImageOps.exif_transpose(image_in)
@@ -59,17 +58,11 @@ def img_concat_h(im1, im2):
     dst.paste(im2, (im1.width, 0))
     return dst
 
-import gradio as gr
-
-def face2vintage(
-    img: Image.Image,
-    size: int
-) -> Image.Image:
-
+def face2vintage(img: Image.Image, size: int) -> Image.Image:
     aligned_img = align_first_face(img)
     if aligned_img is None:
-        output=None
-    else:        
+        output = None
+    else:
         output = session.run([output_name], {input_name: aligned_img})[0]
         output = array_to_image(output)
         aligned_img = array_to_image(aligned_img).resize((output.width, output.height))
@@ -80,13 +73,13 @@ def face2vintage(
 def inference(img):
     out = face2vintage(img, 512)
     return out
-      
-  
+
 title = "Vintage style Pix2PixHD"
-description = "Style a face to look more \"Vintage\". Upload an image with a face, or click on one of the examples below. If a face could not be detected, an image will still be created."
-article = "<hr><p style='text-align: center'>See the <a href='https://github.com/justinpinkney/pix2pixHD' target='_blank'>Github Repo</a></p><p style='text-align: center'>samples: <img src='https://hf.space/gradioiframe/Norod78/VintageStyle/file/Sample00001.jpg' alt='Sample00001'/><img src='https://hf.space/gradioiframe/Norod78/VintageStyle/file/Sample00002.jpg' alt='Sample00002'/><img src='https://hf.space/gradioiframe/Norod78/VintageStyle/file/Sample00003.jpg' alt='Sample00003'/><img src='https://hf.space/gradioiframe/Norod78/VintageStyle/file/Sample00004.jpg' alt='Sample00004'/><img src='https://hf.space/gradioiframe/Norod78/VintageStyle/file/Sample00005.jpg' alt='Sample00005'/></p><p>The \"Vintage Style\" Pix2PixHD model was trained by <a href='https://linktr.ee/Norod78' target='_blank'>Doron Adler</a></p>"
+description = "Style a face to look more \"Vintage\". Upload an image with a face, or click on one of the examples below."
+article = "<hr><p style='text-align: center'>See the <a href='https://github.com/justinpinkney/pix2pixHD' target='_blank'>Github Repo</a></p>"
+
+examples = [['Example00001.jpg'], ['Example00002.jpg'], ['Example00003.jpg'], ['Example00004.jpg'], ['Example00005.jpg']]
 
-examples=[['Example00001.jpg'],['Example00002.jpg'],['Example00003.jpg'],['Example00004.jpg'],['Example00005.jpg'], ['Example00006.jpg']]
 demo = gr.Interface(
     inference, 
     inputs=[gr.Image(type="pil", label="Input")],
@@ -94,10 +87,8 @@ demo = gr.Interface(
     title=title,
     description=description,
     article=article,
-    examples=examples,
-    allow_flagging=False
-    )
-
+    
+)
 
 demo.queue()
 demo.launch()