upgrade1

app.py

@@ -116,7 +116,8 @@ def sepia(input_img):
     pred_img = pred_img.astype(np.uint8)
 
     fig = draw_plot(pred_img, seg)
-    return fig
+    filename = input_image.name if hasattr(input_image, "name") else "Unknown filename"
+    return fig, filename
 
 demo = gr.Interface(fn=sepia,
                     inputs=gr.Image(shape=(400, 600)),