Update app.py

app.py

@@ -32,7 +32,7 @@ pipe = StableDiffusion3Pipeline.from_pretrained(
     model_path, 
     transformer=transformer, 
     torch_dtype=torch.bfloat16
-).to("cuda")
+)
 
 pipe.init_ipadapter(
     ip_adapter_path=ipadapter_path, 
@@ -58,6 +58,7 @@ def process_image(
     height,
     progress=gr.Progress(track_tqdm=True),
 ):
+    pipe.to("cuda")
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)