Update app.py

app.py

@@ -37,7 +37,16 @@ def clear_images():
     return None, None
 
 # Gradio 인터페이스 구성
-with gr.Blocks() as demo:
+with gr.Blocks(css="""
+    #btn-remove {
+        background-color: #4CAF50; /* 초록색 */
+        color: white;
+    }
+    #btn-reset {
+        background-color: #f44336; /* 빨간색 */
+        color: white;
+    }
+""") as demo:
     gr.Markdown("# 이미지 배경 제거기")
     gr.Markdown("이미지를 업로드하면 배경이 자동으로 제거됩니다.")
 
@@ -55,11 +64,11 @@ with gr.Blocks() as demo:
     
     # 배경 제거 버튼
     with gr.Row():
-        btn_remove = gr.Button("배경 제거")
+        btn_remove = gr.Button("배경 제거", elem_id="btn-remove")  # elem_id로 버튼 스타일 지정
         btn_remove.click(fn=remove_background, inputs=input_image, outputs=output_image)
         
         # 초기화 버튼 추가
-        btn_reset = gr.Button("초기화")
+        btn_reset = gr.Button("초기화", elem_id="btn-reset")  # elem_id로 버튼 스타일 지정
         btn_reset.click(fn=clear_images, inputs=None, outputs=[input_image, output_image])
     
     # 예제 이미지
@@ -72,7 +81,5 @@ with gr.Blocks() as demo:
         label="예제 이미지"
     )
 
-    # gr.Markdown("© 2024 이미지 배경 제거기")
-
 if __name__ == "__main__":
     demo.launch()