Update app.py

app.py

@@ -38,6 +38,7 @@ from huggingface_hub import snapshot_download
 
 from model import StableMultiDiffusionPipeline
 from util import seed_everything
+from share_btn import share_js
 
 
 ### Utils
@@ -430,6 +431,21 @@ css = f"""
 .layer-wrap {{
     display: none;
 }}
+
+#share-btn {{
+    color: #ffffff;font-weight: 600;
+    background-color: #000000;
+    font-family: 'IBM Plex Sans', sans-serif;
+    border-radius: 9999px !important;
+}}
+
+#share-btn:hover {{
+    color: #ffffff;font-weight: 600;
+    background-color: #000000;
+    font-family: 'IBM Plex Sans', sans-serif;
+    border-radius: 9999px !important;
+    box-shadow: 0 0 20px #eee;
+}}
 """
 
 for i in range(opt.max_palettes + 1):
@@ -663,6 +679,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
                         # scale=1,
                         elem_id='run-button'
                     )
+                    iface.btn_share = gr.Button('🤗 Share with Community', elem_id='share-btn')
 
                     iface.model_select = gr.Radio(
                         list(model_dict.keys()),
@@ -815,6 +832,8 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
         api_name='model_select',
     )
 
+    iface.btn_share.click(None, [], [], js=share_js)
+
     iface.btn_export_state.click(lambda x: vars(x), state, iface.json_state_export)
     iface.btn_import_state.click(import_state, [state, iface.tbox_state_import], [
         state,