Update app.py

app.py

@@ -9,7 +9,7 @@ from PIL import Image
 from diffusers import DiffusionPipeline
 
 # Define the device
-device = "cuda" if torch.cuda.is_available() else "cpu"
+device = "cuda"
 
 # Use the 'waffles' environment variable as the access token
 hf_token = os.getenv('waffles')
@@ -92,7 +92,7 @@ def run_lora(prompt, cfg_scale, steps, selected_repo, randomize_seed, seed, widt
             seed = random.randint(0, MAX_SEED)
     
     images = generate_images(prompt, trigger_word, steps, seed, cfg_scale, width, height, lora_scale, num_images, progress)
-    pipe.to("cpu")
+    pipe.to("cuda")
     pipe.unload_lora_weights()
     return images, seed  
 
@@ -125,7 +125,7 @@ css = '''
 '''
 with gr.Blocks(theme=gr.themes.Soft(), css=css) as app:
     title = gr.HTML(
-        """<h1><img src="https://huggingface.co/spaces/multimodalart/flux-lora-the-explorer/resolve/main/flux_lora.png" alt="LoRA"> FLUX LoRA the Explorer</h1>""",
+        """<h1><img src="https://huggingface.co/spaces/multimodalart/flux-lora-the-explorer/resolve/main/flux_lora.png" alt="LoRA"> SOONFACTORY over FLUX Schell LoRAs</h1>""",
         elem_id="title",
     )