Update app.py

app.py

@@ -97,6 +97,11 @@ def predict(img):
 # ============================
 # Gradio UI (Webcam + Upload)
 # ============================
+ds = load_dataset("TristanKE/RemainingLifespanPredictionFaces", split="train")
+sample = ds[1]
+img, target = sample["image"], sample["remaining_lifespan"]
+img.save("example.jpg") 
+
 demo = gr.Interface(
     fn=predict,
     inputs=gr.Image(type="pil", sources=["upload", "webcam"], label="Face Photo"),