Upload 2 files

app.py

@@ -0,0 +1,28 @@
+
+import gradio as gr
+from PIL import Image
+import torch
+from ic_light.infer import infer_ic_light  # 假设IC-Light提供一个infer方法
+
+def run_ic_light(foreground_img, background_img, model_type="fbc+fc"):
+    result = infer_ic_light(
+        foreground_path=foreground_img.name,
+        background_path=background_img.name,
+        model_type=model_type
+    )
+    return Image.open(result)
+
+iface = gr.Interface(
+    fn=run_ic_light,
+    inputs=[
+        gr.Image(type="file", label="Foreground Image"),
+        gr.Image(type="file", label="Background Image"),
+        gr.Radio(choices=["fc", "fbc", "fbc+fc"], label="Model Type", value="fbc+fc")
+    ],
+    outputs=gr.Image(label="Relit Image"),
+    title="IC-Light Background Conditional Relighting",
+    description="Upload a foreground and a background image to perform IC-Light based relighting using background conditions."
+)
+
+if __name__ == "__main__":
+    iface.launch()

requirements.txt

@@ -0,0 +1,12 @@
+
+gradio==3.41.2
+torch
+Pillow
+opencv-python
+einops
+transformers==4.36.2
+diffusers==0.27.2
+safetensors
+peft
+protobuf==3.20
+git+https://github.com/lllyasviel/IC-Light.git