.

app.py

@@ -31,8 +31,18 @@ def read_content(file_path: str) -> str:
     return content
 
 def predict(dict, reference, scale, seed, step):
-    init_image = dict["image"].convert("RGB").resize((512,512))
-    mask = dict["mask"].convert("RGB").resize((512,512))
+    width,height=dict["image"].size
+    if width<height:
+        factor=width/512.0
+        width=512
+        height=int((height/factor)/8.0)*8
+
+    else:
+        factor=height/512.0
+        height=512
+        width=int((width/factor)/8.0)*8
+    init_image = dict["image"].convert("RGB").resize((width,height))
+    mask = dict["mask"].convert("RGB").resize((width,height))
     generator = torch.Generator('cuda').manual_seed(seed) if seed != 0 else None
     output = pipe(
         image=init_image,
@@ -118,13 +128,9 @@ with image_blocks as demo:
                             full_width=True,
                         )
                     with gr.Group(elem_id="share-btn-container"):
-                        community_icon = gr.HTML(community_icon_html, visible=False)
-                        loading_icon = gr.HTML(loading_icon_html, visible=False)
-                        share_button = gr.Button("Share to community", elem_id="share-btn", visible=False).style(
-                            margin=False,
-                            rounded=(False, True, True, False),
-                            full_width=True,
-                        )
+                        community_icon = gr.HTML(community_icon_html, visible=True)
+                        loading_icon = gr.HTML(loading_icon_html, visible=True)
+                        share_button = gr.Button("Share to community", elem_id="share-btn", visible=True)
             
             with gr.Row():
                 gr.Examples([