Update app.py

app.py

@@ -76,23 +76,24 @@ def get_gallery():
 
   return images
 
-with gr.Blocks(css=STYLES, elem_id="container") as demo:
-  gr.Markdown("## LLaMA2 Story Showcase", elem_classes=['center'])
-  gr.Markdown("This space is where community shares generated stories by [chansung/co-write-with-llama2](https://huggingface.co/spaces/chansung/co-write-with-llama2) space. "
-              "Generated stories are archived in [chansung/llama2-stories](https://huggingface.co/datasets/chansung/llama2-stories) dataset repository. The gallery will be "
-              "regularly updated in a daily basis.",
-              elem_classes=['small-big', 'center'])
+with gr.Blocks(css=STYLES) as demo:
+  with gr.Column(elem_id="container"):
+    gr.Markdown("## LLaMA2 Story Showcase", elem_classes=['center'])
+    gr.Markdown("This space is where community shares generated stories by [chansung/co-write-with-llama2](https://huggingface.co/spaces/chansung/co-write-with-llama2) space. "
+                "Generated stories are archived in [chansung/llama2-stories](https://huggingface.co/datasets/chansung/llama2-stories) dataset repository. The gallery will be "
+                "regularly updated in a daily basis.",
+                elem_classes=['small-big', 'center'])
     
-  gallery = gr.Gallery(get_gallery, every=3000, columns=5, container=False, elem_id="gallery")
+    gallery = gr.Gallery(get_gallery, every=3000, columns=5, container=False, elem_id="gallery")
 
-  with gr.Column():
-      title = gr.Markdown("title", visible=False, elem_classes=['center'])
-      story = gr.Markdown("stories goes here...", visible=False, elem_classes=['small-big'])
+    with gr.Column():
+        title = gr.Markdown("title", visible=False, elem_classes=['center'])
+        story = gr.Markdown("stories goes here...", visible=False, elem_classes=['small-big'])
 
-  gallery.select(
-      fn=gallery_select,
-      inputs=[gallery],
-      outputs=[title, story]
-  )
+    gallery.select(
+        fn=gallery_select,
+        inputs=[gallery],
+        outputs=[title, story]
+    )
 
 demo.launch()