Update app.py

app.py

@@ -3,6 +3,8 @@ import torch
 from diffusers import StableDiffusionXLPipeline, AutoencoderKL
 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
 import copy
 import json
@@ -32,7 +34,7 @@ with open("sdxl_loras.json", "r") as file:
         for item in data
     ]
 
-device = "cuda"
+device = "cuda" 
 
 state_dicts = {}
 
@@ -98,7 +100,6 @@ def update_selection(selected_state: gr.SelectData, sdxl_loras):
         use_with_diffusers += f"This LoRA is not compatible with diffusers natively yet. But you can still use it on diffusers with sdxl-cog `TokenEmbeddingsHandler` class, check out the [model repo](https://huggingface.co/{lora_repo}#inference-with-🧨-diffusers)"
     use_with_uis = f'''
     ## Use it with Comfy UI, Invoke AI, SD.Next, AUTO1111: 
-
     ### Download the `*.safetensors` weights of [here](https://huggingface.co/{lora_repo}/resolve/main/{weight_name})
     
     - [ComfyUI guide](https://comfyanonymous.github.io/ComfyUI_examples/lora/)
@@ -245,9 +246,9 @@ with gr.Blocks(css="custom.css") as demo:
                 prompt = gr.Textbox(label="Prompt", show_label=False, lines=1, max_lines=1, placeholder="Type a prompt after selecting a LoRA", elem_id="prompt")
                 button = gr.Button("Run", elem_id="run_button")
             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")
+                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")
             result = gr.Image(
                 interactive=False, label="Generated Image", elem_id="result-image"
             )
@@ -313,7 +314,7 @@ with gr.Blocks(css="custom.css") as demo:
     ).success(
         fn=run_lora,
         inputs=[prompt, negative, weight, selected_state, gr_sdxl_loras],
-        outputs=[result],
+        outputs=[result, share_group],
     )
     button.click(
         fn=check_selected,
@@ -323,9 +324,9 @@ with gr.Blocks(css="custom.css") as demo:
     ).success(
         fn=run_lora,
         inputs=[prompt, negative, weight, selected_state, gr_sdxl_loras],
-        outputs=[result],
+        outputs=[result, share_group],
     )
-    # share_button.click(None, [], [], _js=share_js)
+    share_button.click(None, [], [], _js=share_js)
     demo.load(fn=shuffle_gallery, inputs=[gr_sdxl_loras], outputs=[gallery, gr_sdxl_loras], queue=False)
 demo.queue(max_size=20)
 demo.launch(debug=True, auth=custom_auth)