Update app.py

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=10, do_sample=True)
+        generated_ids = hug_model.generate(**model_inputs, max_length=1400, temperature=1, num_return_sequences=12, do_sample=True)
         responses = hug_tokenizer.batch_decode(generated_ids, skip_special_tokens=True)
         codes = []
         for response in responses:
@@ -443,7 +443,7 @@ def main():
             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")
-            with gr.Column(scale=2):
+            with gr.Column(scale=4):
                 output_gallery = gr.Gallery(
             label="Generated images", show_label=False, elem_id="gallery"
         , columns=[5], rows=[2], object_fit="contain", height="auto")