Update README.md

README.md

@@ -66,12 +66,14 @@ from io import BytesIO
 import matplotlib.pyplot as plt
 from PIL import Image, ImageOps
 from diffusers import DiffusionPipeline
-
+import sys
+# CLONE: https://github.com/adobe-research/pattern-analogies
+sys.path.insert(0, "PATH_TO_REPOSITORY")
 SEED = 1729
 DEVICE = th.device("cuda")
 DTYPE = th.float16
 FIG_K = 3
-EXAMPLE_ID = 0
+EXAMPLE_ID = 1
 
 # Now we need to do the trick
 pretrained_path = "bardofcodes/pattern_analogies"
@@ -104,15 +106,15 @@ plt.rcParams['legend.fontsize'] = 'large'
 for i in range(6):
     if i == 0:
         plt.subplot(2, 3, i+1)
-        val_image = img1
+        val_image = images[0]
         label_str = "A"
     elif i == 1:
         plt.subplot(2, 3, i+1)
-        val_image = alt_img1
+        val_image = images[1]
         label_str = "A*"
     elif i == 2:
         plt.subplot(2, 3, i+1)
-        val_image = img2
+        val_image = images[2]
         label_str = "Target"
     else:
         plt.subplot(2, 3,i + 1)
@@ -125,6 +127,7 @@ for i in range(6):
     plt.legend(loc="lower right")
     plt.axis('off')
 plt.subplots_adjust(wspace=0.01, hspace=0.01)
+
 ```
 
 ### Full GitHub Repository