Add example images and improve model download logging in app.py

app.py

@@ -15,12 +15,11 @@ os.makedirs("model", exist_ok=True)
 if not os.path.exists(MODEL_PATH):
     print("Starting model download at", time.strftime("%Y-%m-%d %H:%M:%S"))
     path = hf_hub_download(repo_id="Robys01/face-aging", filename="best_unet_model.pth", local_dir="model", cache_dir="model")
-    print("Model downloaded to", MODEL_PATH, "at", time.strftime("%Y-%m-%d %H:%M:%S"))
+    print(f"Model downloaded to {path}")
 
 model = UNet()  
 model.load_state_dict(torch.load(MODEL_PATH, map_location=torch.device("cpu"), weights_only=False))
 model.eval()
-print("Model loaded at", time.strftime("%Y-%m-%d %H:%M:%S"))
 
 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.
@@ -39,6 +38,10 @@ iface = gr.Interface(
         gr.Slider(10, 90, value=70, step=1, label="Target age", info="Choose the desired age")
     ],
     outputs=gr.Image(type="pil", label="Aged Image"),
+    examples=[
+        ["examples/girl.jpg", 14, 50],
+        ["examples/trump.jpg", 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."
 )