Add submit button and change watermark dimensions

gradio_demo/app.py

@@ -49,7 +49,7 @@ controlnet_path = f'./checkpoints/ControlNetModel'
 controlnet = ControlNetModel.from_pretrained(controlnet_path, torch_dtype=dtype)
 
 logo = Image.open("./gradio_demo/watermark.png")
-logo = logo.resize((100, 70))
+logo = logo.resize((100, 100))
 
 from cv2 import imencode
 import base64
@@ -422,7 +422,7 @@ def main(pretrained_model_name_or_path="wangqixun/YamerMIX_v8", enable_lcm_arg=F
     justify-content: center;
     align-items: center;
     width: 100%;}
-    #store-btn {
+    sub-btn,#store-btn {
     background: linear-gradient(to right, #ffffff, #f2bb13); !important;
     color: #254977 !important;
     }
@@ -439,7 +439,7 @@ def main(pretrained_model_name_or_path="wangqixun/YamerMIX_v8", enable_lcm_arg=F
             with gr.Row(equal_height=True):  # Center the face file
                 with gr.Column(elem_id="centered-face", elem_classes=["centered-column"]):  # Use CSS class for centering
                     face_file = gr.Image(label="Upload a photo of your face", type="pil", height=400, width=500)
-            # submit = gr.Button("Submit", variant="primary")
+            submit = gr.Button("Submit", variant="primary",elem_id="sub-btn")
             with gr.Column():
                 with gr.Row():
                     gallery1 = gr.Image(label="Generated Images")
@@ -499,34 +499,34 @@ def main(pretrained_model_name_or_path="wangqixun/YamerMIX_v8", enable_lcm_arg=F
                 inputs=[face_file,style],
                 outputs=[gallery4]
             )
-#             submit.click(
-#                 fn=remove_tips,
-#                 outputs=usage_tips,
-#                 queue=True,
-#                 api_name=False,
-#                 show_progress = "full"
-#             ).then(
-#                 fn=run_for_prompts1,
-#                 inputs=[face_file,style],
-#                 outputs=[gallery1]
-#             ).then(
-#                 fn=run_for_prompts2,
-#                 inputs=[face_file,style],
-#                 outputs=[gallery2]
-#             ).then(
-#                 fn=run_for_prompts3,
-#                 inputs=[face_file,style],
-#                 outputs=[gallery3]
-#             ).then(
-#                 fn=run_for_prompts4,
-#                 inputs=[face_file,style],
-#                 outputs=[gallery4]
-#             )
+            submit.click(
+                fn=remove_tips,
+                outputs=usage_tips,
+                queue=True,
+                api_name=False,
+                show_progress = "full"
+            ).then(
+                fn=run_for_prompts1,
+                inputs=[face_file,style],
+                outputs=[gallery1]
+            ).then(
+                fn=run_for_prompts2,
+                inputs=[face_file,style],
+                outputs=[gallery2]
+            ).then(
+                fn=run_for_prompts3,
+                inputs=[face_file,style],
+                outputs=[gallery3]
+            ).then(
+                fn=run_for_prompts4,
+                inputs=[face_file,style],
+                outputs=[gallery4]
+            )
             
-#             submit1.click(
-#                 fn=store_images,
-#                 inputs=[email,gallery1,gallery2,gallery3,gallery4,consent],
-#                 outputs=None)
+            submit1.click(
+                fn=store_images,
+                inputs=[email,gallery1,gallery2,gallery3,gallery4,consent],
+                outputs=None)