update

app.py

@@ -15,6 +15,7 @@ from common import (
     preprocess_sam_image_fn,
     select_point,
     start_session,
+    inject_image_css,
 )
 from gradio.themes import Default
 from gradio.themes.utils.colors import slate
@@ -22,6 +23,7 @@ from gradio.themes.utils.colors import slate
 with gr.Blocks(
     delete_cache=(43200, 43200), theme=Default(primary_hue=slate)
 ) as demo:
+    inject_image_css()
     gr.Markdown(
         f"""
         ## Image to 3D Asset Pipeline \n
@@ -41,6 +43,7 @@ with gr.Blocks(
                         image_mode="RGBA",
                         type="pil",
                         height=300,
+                        elem_classes=["image-container"], 
                     )
                     gr.Markdown(
                         """

common.py

@@ -127,6 +127,20 @@ elif os.getenv("GRADIO_APP") == "texture_edit":
 os.makedirs(TMP_DIR, exist_ok=True)
 
 
+def inject_image_css():
+    return gr.HTML(
+        """
+        <style>
+        .image-container img {
+            object-fit: contain !important;
+            max-width: 100% !important;
+            max-height: 100% !important;
+        }
+        </style>
+        """
+    )
+    
+
 def start_session(req: gr.Request) -> None:
     user_dir = os.path.join(TMP_DIR, str(req.session_hash))
     os.makedirs(user_dir, exist_ok=True)