arxivgpt kim commited on
Update app.py
Browse files
app.py
CHANGED
|
@@ -127,22 +127,26 @@ def merge_images(background_path, foreground_image):
|
|
| 127 |
|
| 128 |
return background
|
| 129 |
|
| 130 |
-
# process 함수 수정
|
| 131 |
def process(image, background_image=None):
|
| 132 |
"""
|
| 133 |
이미지에서 배경을 제거합니다. 선택적으로, 사용자가 배경 이미지를 제공한 경우
|
| 134 |
배경 이미지 위에 배경이 제거된 이미지를 투명하게 삽입합니다.
|
| 135 |
"""
|
| 136 |
-
#
|
| 137 |
-
|
|
|
|
|
|
|
|
|
|
| 138 |
# 배경 이미지가 제공되었는지 확인
|
| 139 |
-
if background_image is not None
|
|
|
|
| 140 |
result_image = merge_images(background_image, new_im)
|
| 141 |
else:
|
| 142 |
result_image = new_im
|
| 143 |
|
| 144 |
return result_image
|
| 145 |
|
|
|
|
| 146 |
# Gradio 인터페이스 정의
|
| 147 |
demo = gr.Interface(
|
| 148 |
fn=process,
|
|
|
|
| 127 |
|
| 128 |
return background
|
| 129 |
|
|
|
|
| 130 |
def process(image, background_image=None):
|
| 131 |
"""
|
| 132 |
이미지에서 배경을 제거합니다. 선택적으로, 사용자가 배경 이미지를 제공한 경우
|
| 133 |
배경 이미지 위에 배경이 제거된 이미지를 투명하게 삽입합니다.
|
| 134 |
"""
|
| 135 |
+
# 배경 제거 로직을 여기에 구현합니다...
|
| 136 |
+
# 예를 들어, 배경 제거 처리 후 이미지를 new_im에 할당합니다.
|
| 137 |
+
# 이 예제에서는 단순화를 위해 image를 직접 사용합니다.
|
| 138 |
+
new_im = image # 실제 구현에서는 여기에 배경 제거 로직의 결과를 할당해야 합니다.
|
| 139 |
+
|
| 140 |
# 배경 이미지가 제공되었는지 확인
|
| 141 |
+
if background_image is not None:
|
| 142 |
+
# 배경 이미지에 배경이 제거된 이미지를 삽입하는 로직을 구현합니다.
|
| 143 |
result_image = merge_images(background_image, new_im)
|
| 144 |
else:
|
| 145 |
result_image = new_im
|
| 146 |
|
| 147 |
return result_image
|
| 148 |
|
| 149 |
+
|
| 150 |
# Gradio 인터페이스 정의
|
| 151 |
demo = gr.Interface(
|
| 152 |
fn=process,
|