Update app.py

app.py

@@ -60,9 +60,18 @@ def process_image(image, num_steps=10, guidance_scale=1.0):
     except Exception as e:
         return None, None, f"❌ Error: {str(e)}"
 
+# css = """
+# .image-container img {
+#     object-fit: contain !important;
+# }
+# """
+
 css = """
-.image-container img {
+.image-container img,
+.image-container .fixed img,
+.image-container .fixed {
     object-fit: contain !important;
+    max-height: 400px !important;
 }
 """