Update app.py

app.py

@@ -1,7 +1,6 @@
 import gradio as gr
 from gradio_imageslider import ImageSlider
 from loadimg import load_img
-import spaces
 from transformers import AutoModelForImageSegmentation
 import torch
 from torchvision import transforms
@@ -49,13 +48,13 @@ def fn(image):
     
     return [processed_image], jpg_path  # ImageSlider는 리스트를 기대함
 
-def process_file(f):
-    name_path = f.rsplit(".", 1)[0] + ".png"
-    im = load_img(f, output_type="pil")
-    im = im.convert("RGB")
-    transparent = process(im)
-    transparent.save(name_path)
-    return name_path
+def convert_to_jpg(image):
+    if image is None:
+        return None
+    jpg_image = image.convert("RGB")
+    jpg_path = "downloaded_output.jpg"
+    jpg_image.save(jpg_path, format="JPEG")
+    return jpg_path
 
 # Gradio 컴포넌트 정의
 slider1 = ImageSlider(label="Processed Image", type="pil")
@@ -63,7 +62,7 @@ image_upload = gr.Image(label="Upload an image")
 output_download = gr.File(label="Download JPG File")
 
 # 새로운 샘플 이미지 추가 (app.py와 동일한 폴더에 위치해야 함)
-sample_images = ["1.png", "2.jpg", "3.png"]
+sample_images = [["1.png"], ["2.jpg"], ["3.png"]]
 
 # Gradio 인터페이스 설정
 tab1 = gr.Interface(