Spaces:

weizmannscience
/

multidiffusion-region-based

Sleeping

multimodalart HF staff commited on Feb 27, 2023

Commit

7120442

1 Parent(s): 5fef8b2

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,10 +10,10 @@ MAX_COLORS = 12
 sd = MultiDiffusion("cuda", "2.0")
 def process_sketch(image, binary_matrixes):
-  high_freq_colors = get_high_freq_colors(image)
   how_many_colors = len(high_freq_colors)
   im2arr = np.array(image) # im2arr.shape: height x width x channel
-  im2arr = color_quantization(im2arr, n_colors=how_many_colors)
   colors_fixed = []
   for color in high_freq_colors:

 sd = MultiDiffusion("cuda", "2.0")
 def process_sketch(image, binary_matrixes):
+  high_freq_colors, image = get_high_freq_colors(image)
   how_many_colors = len(high_freq_colors)
   im2arr = np.array(image) # im2arr.shape: height x width x channel
+  im2arr = color_quantization(im2arr, high_freq_colors)
   colors_fixed = []
   for color in high_freq_colors: