Update README.md

README.md

@@ -1,10 +1,61 @@
 ---
-license: mit
+license: cc
+tags:
+- image-to-image
 ---
 
-# InstructPix2Pix checkpoint finetuned with the [RealEdit](https://peter-sushko.github.io/RealEdit/) dataset
+# REALEDIT: Reddit Edits As a Large-scale Empirical Dataset for Image Transformations
+Project page: https://peter-sushko.github.io/RealEdit/
+Data: https://huggingface.co/datasets/peter-sushko/RealEdit
 
-## 1 · Environment preparation
+<img src='https://instruct-pix2pix.timothybrooks.com/teaser.jpg'/>
+
+
+This is the model introduce in realedit paper. There are 2 ways to run inference: either via Diffusers or original InstructPix2Pix pipeline. 
+
+Option 1: Diffusers library:
+
+Install diffusers, transformers library:
+
+```bash
+pip install diffusers accelerate safetensors transformers
+```
+
+Download weights adapted for diffusers:
+
+WEIGHTS
+
+
+```python
+import PIL
+import requests
+import torch
+from diffusers import StableDiffusionInstructPix2PixPipeline, EulerAncestralDiscreteScheduler
+
+model_id = "timbrooks/instruct-pix2pix"
+pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_id, torch_dtype=torch.float16, safety_checker=None)
+
+CODE TO RUN IT:
+
+
+pipe.to("cuda")
+pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
+
+url = "https://raw.githubusercontent.com/timothybrooks/instruct-pix2pix/main/imgs/example.jpg"
+def download_image(url):
+    image = PIL.Image.open(requests.get(url, stream=True).raw)
+    image = PIL.ImageOps.exif_transpose(image)
+    image = image.convert("RGB")
+    return image
+image = download_image(url)
+
+prompt = "turn him into cyborg"
+images = pipe(prompt, image=image, num_inference_steps=10, image_guidance_scale=1).images
+images[0]
+```
+
+
+Option 2: via InstructPix2Pix pipeline:
 
 Clone the repository and set up the directory structure:
 
@@ -23,8 +74,6 @@ cd checkpoints
 
 Return to the repo root and follow the [InstructPix2Pix installation guide](https://github.com/timothybrooks/instruct-pix2pix) to set up the environment.
 
-## 2 · Running the model
-
 ### Edit a single image
 
 ```bash
@@ -55,4 +104,4 @@ If you find this checkpoint helpful, please cite:
       primaryClass={cs.CV},
       url={https://arxiv.org/abs/2502.03629}, 
 }
-```
+```