handling images' final size in display

app.py

@@ -89,6 +89,13 @@ div#share-btn-container > div {flex-direction: row;background: black;align-items
     align-items: center;
     justify-content: center;
 }
+#output-img img, #image_upload img {
+    object-fit: contain;
+    width: 100%;
+    height: auto;
+    max-width: 512px; /* Set maximum display size */
+    max-height: 512px;
+}
 #prompt-container{margin-top:-18px;}
 #prompt-container .form{border-top-left-radius: 0;border-top-right-radius: 0}
 #image_upload{border-bottom-left-radius: 0px;border-bottom-right-radius: 0px}