Update app.py

app.py

@@ -1,4 +1,3 @@
-import spaces
 import gradio as gr
 import torch
 from PIL import Image
@@ -10,7 +9,7 @@ import numpy as np
 
 def save_image(img):
     unique_name = str(uuid.uuid4()) + ".png"
-    img.save(unique_index)
+    img.save(unique_name)
     return unique_name
 
 def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
@@ -84,8 +83,10 @@ 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
     pipe.load_lora_weights(lora_repo)
-    pipe.to("cuda")
+    pipe.unet.to("cuda")  # Move UNet to GPU
+    pipe.text_encoder.to("cuda")  # Move text encoder to GPU
 
     if trigger_word:
         positive_prompt = f"{trigger_word} {positive_prompt}"
@@ -139,12 +140,6 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
             run_button = gr.Button("Generate as : ( 1280 x 832 )🤗", scale=0, elem_classes="submit-btn")
             
             with gr.Accordion("Advanced options", open=True, visible=True):
-                model_choice = gr.Dropdown(
-                    label="LoRA Selection",
-                    choices=list(LORA_OPTIONS.keys()),
-                    value="Sketch-Smudge"
-                )
-                
                 seed = gr.Slider(
                     label="Seed",
                     minimum=0,
@@ -196,7 +191,11 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
                     label="Quality Style",
                 )
 
-
+                model_choice = gr.Dropdown(
+                    label="LoRA Selection",
+                    choices=list(LORA_OPTIONS.keys()),
+                    value="Sketch-Smudge"
+                )
         
         with gr.Column(scale=2):
             result = gr.Gallery(label="Result", columns=1, show_label=False)