update

app.py

@@ -224,7 +224,7 @@ def llm_call(question_prompt, model_name,
     top_p=1, n_samples=64, stop=None):
     if HUGGINGFACE:
         model_inputs = hug_tokenizer([question_prompt], return_tensors="pt").to('cuda')
-        generated_ids = hug_model.generate(**model_inputs, max_length=1400, temperature=1, num_return_sequences=9, do_sample=True)
+        generated_ids = hug_model.generate(**model_inputs, max_length=1400, temperature=1, num_return_sequences=10, do_sample=True)
         responses = hug_tokenizer.batch_decode(generated_ids, skip_special_tokens=True)
         codes = []
         for response in responses:
@@ -438,13 +438,14 @@ def main():
         gr.Markdown("""Here we can draw a target image using the sketchpad, and see what kinds of graphics program LLM generates. To allow the LLM to visually perceive the input image, we convert the image to ASCII strings.""")
         gr.Markdown("## Draw logo")
         with gr.Row():
-            with gr.Column():
+            with gr.Column(scale=1):
                 canvas = gr.Sketchpad(canvas_size=(512,512), brush=Brush(colors=["black"], default_size=2, color_mode='fixed'))
                 submit_button = gr.Button("Submit")
-            output_gallery = gr.Gallery(
-        label="Generated images", show_label=False, elem_id="gallery"
-    , columns=[3], rows=[3], object_fit="contain", height="auto")
-            # output_image = gr.Image(label="output")
+            with gr.Column(scale=2):
+                output_gallery = gr.Gallery(
+            label="Generated images", show_label=False, elem_id="gallery"
+        , columns=[5], rows=[2], object_fit="contain", height="auto")
+                # output_image = gr.Image(label="output")
         
         submit_button.click(img_to_code_img, inputs=canvas, outputs=output_gallery)
         demo.load(