Update app.py

app.py

@@ -584,7 +584,7 @@ with gr.Blocks(css=STYLES) as demo:
                       "design your own prompt in the textbox below, or just hit 'generate prompt for cover art` button. LLaMA2 "
                       "model will suggest a prompt for you based on your story.")
 
-          cover_art = gr.Image(interactive=False, elem_classes=['no-label-image'])
+          cover_art = gr.Image(interactive=False, height=256, width=256, elem_classes=['no-label-image'])
           gen_cover_art_prompt = gr.Textbox(lines=3, max_lines=3, elem_classes=['no-label'])
           # toggle between "generate prompt for cover art" and "generate cover art"
           gen_cover_art_btn = gr.Button("generate prompt for cover art", elem_classes=['control-label-font', 'control-button'])