Update app.py

app.py

@@ -1,10 +1,10 @@
 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
+from PIL import Image
 
 torch.set_float32_matmul_precision(["high", "highest"][0])
 
@@ -25,20 +25,18 @@ transform_image = transforms.Compose(
 def fn(image):
     if image is None or len(image) == 0:
         return image, None  # 원본 이미지도 반환
-    im = load_img(image, output_type="pil")
-    im = im.convert("RGB")
+    im = Image.open(image).convert("RGB")
     image_size = im.size
     origin = im.copy()
-    image = load_img(im)
-    input_images = transform_image(image).unsqueeze(0).to("cuda")
+    input_images = transform_image(im).unsqueeze(0).to("cuda")
     # Prediction
     with torch.no_grad():
         preds = birefnet(input_images)[-1].sigmoid().cpu()
     pred = preds[0].squeeze()
     pred_pil = transforms.ToPILImage()(pred)
     mask = pred_pil.resize(image_size)
-    image.putalpha(mask)
-    return image, origin  # 변환된 이미지와 원본 이미지 반환
+    im.putalpha(mask)
+    return im, origin  # 변환된 이미지와 원본 이미지 반환
 
 
 def save_image(image):
@@ -56,23 +54,21 @@ with gr.Blocks() as demo:
     slider1 = ImageSlider(label="birefnet", type="pil")
     slider2 = ImageSlider(label="birefnet", type="pil")
 
-    chameleon = load_img("butterfly.jpg", output_type="pil")
-    example_image2 = load_img("example2.jpg", output_type="pil")  # 두 번째 예제 이미지
-    example_image3 = load_img("example3.jpg", output_type="pil")  # 세 번째 예제 이미지
-    url1 = "https://hips.hearstapps.com/hmg-prod/images/gettyimages-1229892983-square.jpg"
-    url2 = "https://example.com/example2.jpg"  # 두 번째 예제 URL
-    url3 = "https://example.com/example3.jpg"  # 세 번째 예제 URL
+    # 스페이스에 있는 예제 이미지 파일 경로
+    example_image1 = "example_images/example1.jpg"
+    example_image2 = "example_images/example2.jpg"
+    example_image3 = "example_images/example3.jpg"
     
     with gr.Tab("Image Upload"):
         tab1 = gr.Interface(
             fn, inputs=image, outputs=[slider1, output_file], 
-            examples=[chameleon, example_image2, example_image3], api_name="image"
+            examples=[example_image1, example_image2, example_image3], api_name="image"
         )
 
     with gr.Tab("Image URL"):
         tab2 = gr.Interface(
             fn, inputs=text, outputs=[slider2, output_file], 
-            examples=[url1, url2, url3], api_name="text"
+            examples=[example_image1, example_image2, example_image3], api_name="text"
         )
 
     def process_download(image):