Update README.md

README.md

@@ -115,6 +115,7 @@ output_t2i = pipeline(
 for i, img in enumerate(output_t2i["images"]):
     img.save(f"text2image_{i}.png")
 ```
+![Text2image](fig2.png)
 
 ### Image-to-Image Generation
 
@@ -144,6 +145,10 @@ for i, img in enumerate(output_i2i["images"]):
     img.save(f"image2image_{i}.png")
 ```
 
+![Image2image](fig1.png)
+
+
+
 ## Software Dependencies
 
 - torch>=2.0.0