Update app.py

app.py

@@ -1,4 +1,3 @@
-import spaces
 import gradio as gr
 import torch
 from PIL import Image
@@ -84,10 +83,9 @@ def generate(
     lora_repo = LORA_OPTIONS[lora_choice]["lora_repo"]
     trigger_word = LORA_OPTIONS[lora_choice]["trigger_word"]
     
-    # Load LoRA weights and move them to GPU
+    # Load LoRA weights and move the entire pipeline to GPU
     pipe.load_lora_weights(lora_repo)
-    pipe.unet.to("cuda")  # Move UNet to GPU
-    pipe.text_encoder.to("cuda")  # Move text encoder to GPU
+    pipe.to("cuda")  # Move the entire pipeline to GPU
 
     if trigger_word:
         positive_prompt = f"{trigger_word} {positive_prompt}"
@@ -230,4 +228,4 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
     )
 
 if __name__ == "__main__":
-    demo.queue(max_size=40).launch()
+    demo.queue(max_size=40).launch(share=True)  # Set share=True to create a public link