update readme

README.md

@@ -3,17 +3,28 @@ license: mit
 ---
 # Unique3d-MVImage-Diffuser Model Card 
 
+<div align="left">
+  <a href='LICENSE'><img src='https://img.shields.io/badge/license-MIT-yellow'></a> 
+  <a href='https://wukailu.github.io/Unique3D'><img src='https://img.shields.io/badge/Project-Unique3D-green'></a>
+  <a href='https://huggingface.co/Luffuly/unique3d-normal-diffuser'><img src='https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-Normal-blue'></a>
+   <a href='https://huggingface.co/Luffuly/unique3d-mvimage-diffuser'><img src='https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-MVImage-red'></a> 
+  <br>
+</div>
+<br>
+
 ## Example  
-Note the input image is suppose to be **white background**. 
+Note the input image is required to be **white background**. 
+
+![mv-boy](https://github.com/user-attachments/assets/65558519-dbfe-40de-ac13-5d78527541c5)
+
 ```bash 
 import torch 
 import numpy as np 
 from PIL import Image 
 from pipeline import StableDiffusionImage2MVCustomPipeline
 
-
-pipe = StableDiffusionImage2MVCustomPipeline.from_pretrained( 
-    "Luffuly/unique3d-mv-variation-diffuser", 
+pipe = Unique3dDiffusionPipeline.from_pretrained( 
+    "Luffuly/unique3d-mvimage-diffuser", 
     torch_dtype=torch.float16, 
     trust_remote_code=True,  
     class_labels=torch.tensor(range(4)),
@@ -23,7 +34,7 @@ seed = -1
 generator = torch.Generator(device='cuda').manual_seed(-1)
 
 
-image = Image.open('image.png') 
+image = Image.open('data/boy.png') 
 forward_args = dict(
     width=256,
     height=256,
@@ -37,12 +48,11 @@ forward_args = dict(
 
 out = pipe(image, **forward_args).images
 rgb_np = np.hstack([np.array(img) for img in out])
-Image.fromarray(rgb_np).save(f"test.png")
+Image.fromarray(rgb_np).save(f"mv-boy.png")
 
 ```
 
-## Image-to-Normal 
-please check: 
+
 
 ## Citation
 ```bash