Update app.py

app.py

@@ -72,7 +72,7 @@ def infer(prompt):
 css="""
 #col-container {
     margin: 0 auto;
-    max-width: 90vw;
+    max-height: 90vh;
 }
 #image-container {
     aspect-ratio: 1 / 1;