Update app.py

app.py

@@ -9,7 +9,7 @@ import random
 import numpy as np
 import spaces
 import huggingface_hub
-
+import copy
 
 from FlowEdit_utils import FlowEditSD3, FlowEditFLUX
 SD3STRING = 'stabilityai/stable-diffusion-3-medium-diffusers'
@@ -96,15 +96,15 @@ def FlowEditRun(
 
     if model_type == 'FLUX':
         # pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.float16, token=os.getenv('HF_ACCESS_TOK'))
-        pipe = pipe_flux # still on CPU
+        pipe = copy.deepcopy(pipe_flux) # still on CPU
     elif model_type == 'SD3':
         # pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers", torch_dtype=torch.float16, token=os.getenv('HF_ACCESS_TOK'))
-        pipe = pipe_sd3 # still on CPU
+        pipe = copy.deepcopy(pipe_sd3) # still on CPU
     else:
         raise NotImplementedError(f"Model type {model_type} not implemented")
 
         scheduler = pipe.scheduler
-        pipe = pipe.to("cuda")
+        pipe = pipe.to(device)