Update README.md

README.md

@@ -1,11 +1,16 @@
 ---
-license: mit
-tags:
-- image-to-image
+title: ip2p-RoboPredict Demo
+emoji: 🤖
+colorFrom: blue
+colorTo: pink
+sdk: gradio
+sdk_version: "3.41.2"
+app_file: app.py
+pinned: false
 ---
 # ip2p-RoboPredict: InstructPix2Pix Fine-tuning for Robotic Action Frame Prediction
 GitHub: https://github.com/yutengzhang03/ip2p-finetune
-<img src='https://github.com/yutengzhang03/ip2p-finetune/blob/main/img/show-example.png'/>
+<img src='img/show-example.png'/>
 
 
 
@@ -22,11 +27,11 @@ import PIL
 import requests
 import torch
 from diffusers import StableDiffusionInstructPix2PixPipeline, EulerAncestralDiscreteScheduler
-model_id = "timbrooks/instruct-pix2pix"
+model_id = "yutengz/ip2p-RoboPredict"
 pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_id, torch_dtype=torch.float16, safety_checker=None)
 pipe.to("cuda")
 pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
-url = "https://raw.githubusercontent.com/timothybrooks/instruct-pix2pix/main/imgs/example.jpg"
+url = "https://github.com/yutengzhang03/ip2p-finetune/blob/main/img/source.png"
 def download_image(url):
     image = PIL.Image.open(requests.get(url, stream=True).raw)
     image = PIL.ImageOps.exif_transpose(image)
@@ -36,4 +41,4 @@ 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]
-```
+```