Update app.py

app.py

@@ -7,7 +7,7 @@ from region_control import MultiDiffusion, get_views, preprocess_mask
 from sketch_helper import get_high_freq_colors, color_quantization, create_binary_matrix
 MAX_COLORS = 12
 
-sd = MultiDiffusion("cuda", "2.1")
+sd = MultiDiffusion("cuda", "2.0")
 
 def process_sketch(image, binary_matrixes):
   high_freq_colors = get_high_freq_colors(image)
@@ -21,7 +21,7 @@ def process_sketch(image, binary_matrixes):
     if any(c != 255 for c in (r, g, b)):
       binary_matrix = create_binary_matrix(im2arr, (r,g,b))
       binary_matrixes.append(binary_matrix)
-      colors_fixed.append(gr.update(value=f'<div class="color-bg-item" style="background-color: rgb({r},{g},{b})"></div>'))
+      colors_fixed.append(gr.update(value=f'<div><img src="file/{binary_matrix}" /><div class="color-bg-item" style="background-color: rgb({r},{g},{b})"></div></div>'))
   visibilities = []
   colors = []
   for n in range(MAX_COLORS):