Update app.py

app.py

@@ -180,10 +180,10 @@ def infer_with_lora_wrapper(input_image, prompt, selected_index, custom_lora, se
 @spaces.GPU
 @torch.no_grad()
 def infer_with_lora(input_image, prompt, selected_index, custom_lora, seed=42, randomize_seed=False, steps=28, guidance_scale=2.5, lora_scale=1.0, width=960, height=1280, flux_loras=None, progress=gr.Progress(track_tqdm=True)):
-    pipe.to("cuda")
-    generator = torch.Generator(device="cuda").manual_seed(seed)
     """Generate image with selected LoRA"""
     global current_lora, pipe
+    pipe.to("cuda")
+    generator = torch.Generator(device="cuda").manual_seed(seed)
     
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)