Update app.py

app.py

@@ -28,7 +28,7 @@ state_dict = load_state_dict(model_file)
 model, _, _, _, _ = ControlNetModel_Union._load_pretrained_model(
     controlnet_model, state_dict, model_file, "xinsir/controlnet-union-sdxl-1.0"
 )
-model.to(device="cuda", dtype=torch.float16)
+model.to(device="cpu", dtype=torch.float16)
 
 vae = AutoencoderKL.from_pretrained(
     "madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16
@@ -40,7 +40,7 @@ pipe = StableDiffusionXLFillPipeline.from_pretrained(
     vae=vae,
     controlnet=model,
     variant="fp16",
-).to("cuda")
+).to("cpu")
 
 pipe.scheduler = TCDScheduler.from_config(pipe.scheduler.config)
 
@@ -50,10 +50,9 @@ prompt = "high quality"
     negative_prompt_embeds,
     pooled_prompt_embeds,
     negative_pooled_prompt_embeds,
-) = pipe.encode_prompt(prompt, "cuda", True)
+) = pipe.encode_prompt(prompt, "cpu", True)
 
 
-@spaces.GPU(duration=16)
 def fill_image(image, model_selection):
     source = image["background"]
     mask = image["layers"][0]
@@ -103,7 +102,6 @@ with gr.Blocks(css=css) as demo:
         input_image = gr.ImageMask(
             type="pil",
             label="Input Image",
-            crop_size=(1024, 1024),
             canvas_size=(1024, 1024),
             layers=False,
             sources=["upload"],