Upload app.py

app.py

@@ -14,7 +14,7 @@ transformer = FluxTransformer2DModel.from_pretrained(
         "black-forest-labs/FLUX.1-dev", subfolder='transformer', torch_dytpe=torch.bfloat16
     )
 
-mps_device =torch.device("mps")
+cuda_device =torch.device("cuda")
 # Build pipeline
 controlnet = FluxControlNetModel.from_pretrained("alimama-creative/FLUX.1-dev-Controlnet-Inpainting-Beta", torch_dtype=torch.bfloat16)
 pipe = FluxControlNetInpaintingPipeline.from_pretrained(
@@ -22,7 +22,7 @@ pipe = FluxControlNetInpaintingPipeline.from_pretrained(
     controlnet=controlnet,
     transformer=transformer,
     torch_dtype=torch.bfloat16
-).to(mps_device)
+).to(cuda_device)
 pipe.transformer.to(torch.bfloat16)
 pipe.controlnet.to(torch.bfloat16)
 
@@ -49,7 +49,7 @@ def process(input_image_editor,
     
     image = image.convert("RGB").resize(size)
     mask = mask.convert("RGB").resize(size)
-    generator = torch.Generator(device="mps").manual_seed(seed)
+    generator = torch.Generator(device="cuda").manual_seed(seed)
     result = pipe(
     prompt=prompt,
     height=size[1],