Update app.py

app.py

@@ -91,7 +91,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("gpu")
+    pipe.to("cpu")
     pipe.unload_lora_weights()
     return images, seed