Update app.py

seg_map -> overlay

app.py

@@ -255,7 +255,7 @@ def process_coastal_image(location, input_image):
     # Try uploading to DigitalOcean Spaces
     url = "Local Storage"
     try:
-        url = upload_mask(Image.fromarray(seg_map), prefix=location.replace(' ', '_'))
+        url = upload_mask(Image.fromarray(overlay), prefix=location.replace(' ', '_'))
     except Exception as e:
         print(f"Upload failed: {e}")
         url = f"Upload error: {str(e)}"
@@ -291,7 +291,7 @@ def process_with_alignment(location, reference_image, input_image):
     # Try uploading to DigitalOcean Spaces
     url = "Local Storage"
     try:
-        url = upload_mask(Image.fromarray(seg_map), prefix="aligned_" + location.replace(' ', '_'))
+        url = upload_mask(Image.fromarray(overlay), prefix="aligned_" + location.replace(' ', '_'))
     except Exception as e:
         print(f"Upload failed: {e}")
         url = f"Upload error: {str(e)}"