Update app.py

app.py

@@ -99,16 +99,25 @@ INV_AGE_SCALE = 80  # training used age/80 normalization
 
 
 def draw_caption_on_image(image, caption):
+    if image.mode != "RGBA":
+        image = image.convert("RGBA")
     draw = ImageDraw.Draw(image)
-    font = ImageFont.load_default()  # Default font (always available)
+    font = ImageFont.load_default()
     
     bbox = draw.textbbox((0,0), caption, font=font)
     text_w = bbox[2] - bbox[0]
     text_h = bbox[3] - bbox[1]
-    
-    draw.rectangle([0,0,text_w+20,text_h+20], fill=(0,0,0,127))
+
+    # semi-transparent rectangle
+    overlay = Image.new("RGBA", image.size)
+    overlay_draw = ImageDraw.Draw(overlay)
+    overlay_draw.rectangle([0,0,text_w+20,text_h+20], fill=(0,0,0,127))
+    image = Image.alpha_composite(image, overlay)
+
+    draw = ImageDraw.Draw(image)
     draw.text((10,10), caption, font=font, fill="white")
-    return image
+    return image.convert("RGB")
+
 
 
 # --- Prediction function for multiple images ---