Update app.py

app.py

@@ -35,11 +35,12 @@ def infer(image_input, audience, keyword, protagonist):
         chapter_story = chat_completion.choices[0].message.content
         chapters.append(chapter_story)
         images.append(current_image_url)
-
-    return {
-        "chapter1_story": chapters[0],
-        "chapter1_image": images[0],
-    }, gr.Group.update(visible=True)
+        
+    return chapters[0], images[0]
+    # return {
+    #     "chapter1_story": chapters[0],
+    #     "chapter1_image": images[0],
+    # }, gr.Group.update(visible=True)
 
 css = """
 #col-container {max-width: 910px; margin-left: auto; margin-right: auto;}
@@ -64,7 +65,6 @@ with gr.Blocks(css=css) as demo:
             with gr.Column():
                 chapter1_story = gr.Markdown(label="Chapter 1: 이야기", elem_id="chapter1_story")
                 chapter1_image = gr.Image(label="Chapter 1: 그림", elem_id="chapter1_image")
-
     submit_btn.click(fn=infer, inputs=[image_in, audience, keyword_in, protagonist_in], 
                      outputs=[chapter1_story, chapter1_image])
     demo.queue(max_size=12).launch()