fix

app.py

@@ -94,12 +94,13 @@ def remove_background(image_path, mask):
     
     return result
     
-pipe = FluxInpaintPipeline.from_pretrained(bfl_repo, torch_dtype=torch.bfloat16).to(DEVICE)
+pipe = FluxInpaintPipeline.from_pretrained(bfl_repo, torch_dtype=torch.bfloat16)
 MAX_SEED = np.iinfo(np.int32).max
 TRIGGER = "a photo of TOK"
 
 @spaces.GPU(duration=200)
 def execute(image, prompt):
+  pipe.to(DEVICE)
   if not prompt :
         gr.Info("Please enter a text prompt.")
         return None