Add Replicate demo and API support

.dockerignore

@@ -0,0 +1,21 @@
+# The .dockerignore file excludes files from the container build process.
+#
+# https://docs.docker.com/engine/reference/builder/#dockerignore-file
+
+# Cog
+/models/*
+/output_images/*
+
+# Exclude Git files
+.git
+.github
+.gitignore
+
+# Exclude Python cache files
+__pycache__
+.mypy_cache
+.pytest_cache
+.ruff_cache
+
+# Exclude Python virtual environment
+/venv

.gitignore

@@ -160,3 +160,12 @@ cython_debug/
 #  and can be added to the global gitignore or merged into this file.  For a more nuclear
 #  option (not recommended) you can uncomment the following to ignore the entire idea folder.
 .idea/
+
+# Cog
+/models/*
+/output_images/*
+/.cog/
+output.*.png
+output.*.webp
+output.*.jpg
+output.*.jpeg

README.md

@@ -1,5 +1,8 @@
 # IC-Light
 
+> [![Replicate - Text-Conditioned Model](https://replicate.com/zsxkib/ic-light/badge)](https://replicate.com/zsxkib/ic-light) ✍️✨ Prompts to auto-magically relight your images!
+> [![Replicate - Background-Conditioned Model](https://replicate.com/zsxkib/ic-light-background/badge)](https://replicate.com/zsxkib/ic-light-background) 🖼️✨ Background images + prompts to auto-magically relight your images!
+
 IC-Light is a project to manipulate the illumination of images.
 
 The name "IC-Light" stands for **"Imposing Consistent Light"** (we will briefly describe this at the end of this page).

cog.yaml

@@ -0,0 +1,36 @@
+# Configuration for Cog ⚙️
+# Reference: https://cog.run/yaml
+
+build:
+  # set to true if your model requires a GPU
+  gpu: true
+  cuda: "12.1"
+
+  # a list of ubuntu apt packages to install
+  system_packages:
+    - "libgl1-mesa-glx"
+    - "libglib2.0-0"
+
+  # python version in the form '3.11' or '3.11.4'
+  python_version: "3.10"
+
+  # a list of packages in the format <package-name>==<version>
+  python_packages:
+    - diffusers==0.27.2
+    - transformers==4.36.2
+    - opencv-python
+    - safetensors
+    - pillow==10.2.0
+    - einops
+    - torch
+    - torchvision
+    - peft
+    - gradio==3.41.2
+    - protobuf==3.20
+
+  # commands run after the environment is setup
+  run:
+    - curl -o /usr/local/bin/pget -L "https://github.com/replicate/pget/releases/download/v0.8.1/pget_linux_x86_64" && chmod +x /usr/local/bin/pget
+
+# predict.py defines how predictions are run on your model
+predict: "predict.py:Predictor"