Update app.py

app.py

@@ -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><img src="file/{binary_matrix}" /><div class="color-bg-item" style="background-color: rgb({r},{g},{b})"></div></div>'))
+      colors_fixed.append(gr.update(value=f'<div style="display:flex;align-items: center;justify-content: center"><img width="20%" 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):
@@ -81,7 +81,7 @@ with gr.Blocks(css=css) as demo:
           with gr.Row(visible=False) as color_row[n]:
             with gr.Box(elem_id="color-bg"):
               colors.append(gr.HTML('<div class="color-bg-item" style="background-color: black"></div>'))
-            prompts.append(gr.Textbox(label="Prompt for this color"))
+            prompts.append(gr.Textbox(label="Prompt for this mask"))
         final_run_btn = gr.Button("Generate!")
     
     out_image = gr.Image(label="Result")