Pre-load example images for the face aging demo to solve gradio move issue.

app.py

@@ -5,7 +5,6 @@ from models import UNet
 from test_functions import process_image
 from PIL import Image
 import gradio as gr
-
 from huggingface_hub import hf_hub_download
 
 MODEL_PATH = "model/best_unet_model.pth"
@@ -22,14 +21,16 @@ model.load_state_dict(torch.load(MODEL_PATH, map_location=torch.device("cpu"), w
 model.eval()
 
 def age_image(image: Image.Image, source_age: int, target_age: int) -> Image.Image:
-    # Ensure the image is in RGB or grayscale; if not, convert it.
     if image.mode not in ["RGB", "L"]:
         print(f"Converting image from {image.mode} to RGB")
         image = image.convert("RGB")
-
     processed_image = process_image(model, image, source_age, target_age)
     return processed_image
 
+# Pre-load the example images as PIL objects
+example1 = Image.open("examples/girl.jpg")
+example2 = Image.open("examples/trump.jpg")
+
 iface = gr.Interface(
     fn=age_image,
     inputs=[
@@ -39,8 +40,8 @@ iface = gr.Interface(
     ],
     outputs=gr.Image(type="pil", label="Aged Image"),
     examples=[
-        ["examples/girl.jpg", 14, 50],
-        ["examples/trump.jpg", 74, 30],
+        [example1, 14, 50],
+        [example2, 74, 30],
     ],
     title="Face Aging Demo",
     description="Upload an image along with a source age approximation and a target age to generate an aged version of the face."