Update app.py

app.py

@@ -118,13 +118,22 @@ def crop(img: Image) -> Image:
 with tempfile.TemporaryDirectory() as tmpdir:
     with gr.Blocks(
         title='StereoGen Demo',
-        css='img {display: inline;}'
+        css="""
+        .badge-container {
+            display: flex;
+            gap: 8px;
+            flex-wrap: wrap;
+            margin: 1em 0;
+        }
+        .badge-container img {
+            height: 28px;
+            display: inline-block;
+        }
+        """
     ) as demo:
         # Internal states.
         src_image = gr.State()
         src_depth = gr.State()
-        proj_mtx = gr.State()
-        src_view_mtx = gr.State()
 
         # Blocks.
         gr.Markdown(