Update app.py

app.py

@@ -5,7 +5,6 @@ import timm
 import time
 from huggingface_hub import hf_hub_download
 from safetensors.torch import load_file
-from share_btn import community_icon_html, loading_icon_html, share_js
 from cog_sdxl_dataset_and_utils import TokenEmbeddingsHandler
 
 import lora
@@ -148,13 +147,7 @@ pipe.to(device)
 
 last_lora = ""
 last_fused = False
-js = '''
-var button = document.getElementById('button');
-// Add a click event listener to the button
-button.addEventListener('click', function() {
-    element.classList.add('selected');
-});
-'''
+
 def update_selection(selected_state: gr.SelectData, sdxl_loras, face_strength, image_strength, weight, depth_control_scale, negative, is_new=False):
     lora_repo = sdxl_loras[selected_state.index]["repo"]
     new_placeholder = "Type a prompt to use your selected LoRA"
@@ -349,7 +342,7 @@ def run_lora(face_image, prompt, negative, lora_scale, selected_state, face_stre
 
     st = time.time()
     image = generate_image(prompt, negative, face_emb, face_image, face_kps, image_strength, guidance_scale, face_strength, depth_control_scale, repo_name, full_path_lora, lora_scale, sdxl_loras, selected_state_index, st)
-    return image, gr.update(visible=True)
+    return image
 
 def shuffle_gallery(sdxl_loras):
     random.shuffle(sdxl_loras)
@@ -368,7 +361,7 @@ def swap_gallery(order, sdxl_loras):
 def deselect():
   return gr.Gallery(selected_index=None)
 
-with gr.Blocks(css="custom.css") as demo:
+with gr.Blocks() as demo:
     gr_sdxl_loras = gr.State(value=sdxl_loras_raw)
     title = gr.HTML(
         """<h1><img src="https://i.imgur.com/DVoGw04.png">
@@ -409,10 +402,7 @@ with gr.Blocks(css="custom.css") as demo:
             result = gr.Image(
                 interactive=False, label="Generated Image", elem_id="result-image"
             )
-            with gr.Group(elem_id="share-btn-container", visible=False) as share_group:
-                community_icon = gr.HTML(community_icon_html)
-                loading_icon = gr.HTML(loading_icon_html)
-                share_button = gr.Button("Share to community", elem_id="share-btn")
+
             with gr.Accordion("Advanced options", open=False):
                 negative = gr.Textbox(label="Negative Prompt")
 
@@ -463,7 +453,7 @@ with gr.Blocks(css="custom.css") as demo:
     ).success(
         fn=run_lora,
         inputs=[photo, prompt, negative, weight, selected_state, face_strength, image_strength, guidance_scale, depth_control_scale, gr_sdxl_loras],
-        outputs=[result, share_group],
+        outputs=[result],
     )
     button.click(
         fn=check_selected,
@@ -473,9 +463,9 @@ with gr.Blocks(css="custom.css") as demo:
     ).success(
         fn=run_lora,
         inputs=[photo, prompt, negative, weight, selected_state, face_strength, image_strength, guidance_scale, depth_control_scale, gr_sdxl_loras],
-        outputs=[result, share_group],
+        outputs=[result],
     )
-    share_button.click(None, [], [], js=share_js)
-    demo.load(fn=classify_gallery, inputs=[gr_sdxl_loras], outputs=[gallery, gr_sdxl_loras], queue=False, js=js)
+
+    demo.load(fn=classify_gallery, inputs=[gr_sdxl_loras], outputs=[gallery, gr_sdxl_loras], queue=False)
 demo.queue(max_size=20)
-demo.launch(share=True)
+demo.launch()