Update app.py

app.py

@@ -25,7 +25,7 @@ def fn(image):
     im = im.convert("RGB")
     origin = im.copy()
     processed_image = process(im)
-    # Convert processed image to JPEG
+    # Convert processed image to JPEG for download
     buffered = BytesIO()
     processed_image.convert("RGB").save(buffered, format="JPEG")
     buffered.seek(0)
@@ -43,27 +43,16 @@ def process(image):
     image.putalpha(mask)
     return image
 
-def process_download(f):
-    name_path = f.rsplit(".", 1)[0] + ".jpg"
-    im = load_img(f, output_type="pil")
-    im = im.convert("RGB")
-    transparent = process(im)
-    # Convert to JPEG
-    buffered = BytesIO()
-    transparent.convert("RGB").save(buffered, format="JPEG")
-    buffered.seek(0)
-    return buffered
-
 slider = ImageSlider(label="Processed Image", type="pil")
 download_output = gr.File(label="Download JPG File")
 
 image_upload = gr.Image(label="Upload an image")
 
-# 새로운 샘플 이미지
+# 새로운 샘플 이미지 (예: 동일 디렉토리에 위치)
 sample_images = [
-    "1.png",
-    "2.jpg",
-    "3.png"
+    ["1.png"],
+    ["2.jpg"],
+    ["3.png"]
 ]
 
 tab = gr.Interface(