bugfix

app.py

@@ -36,7 +36,7 @@ IMAGE_SIZE = 512
 device = "cuda" if torch.cuda.is_available() else "cpu"
 base_model = "black-forest-labs/FLUX.1-dev"
 
-controlnet_model = 'InstantX/FLUX.1-dev-Controlnet-Union'
+controlnet_model = 'InstantX/FLUX.1-dev-Controlnet-Canny'
 controlnet = FluxControlNetModel.from_pretrained(controlnet_model, torch_dtype=torch.bfloat16)
 
 
@@ -158,6 +158,7 @@ def run_flux(
     progress
 ) -> Image.Image:
     print("Running FLUX...")
+    pipe.to(device)
     width, height = resolution_wh
     if randomize_seed_checkbox:
         seed_slicer = random.randint(0, MAX_SEED)
@@ -366,7 +367,7 @@ with gr.Blocks() as demo:
                 )
             
             control_mode = gr.Dropdown(
-                [ "canny", "depth",  "pose"], label="Controlnet Model", info="choose controlnet model!", value="canny"
+                [ "canny"], label="Controlnet Model", info="choose controlnet model!", value="canny"
             )
             lora_strings_json = gr.Text(label="LoRA Configs (JSON List String)", placeholder='[{"repo": "lora_repo1", "weights": "weights1", "adapter_name": "adapter_name1", "adapter_weight": 1}, {"repo": "lora_repo2", "weights": "weights2", "adapter_name": "adapter_name2", "adapter_weight": 1}]', lines=5)
 
@@ -427,7 +428,7 @@ with gr.Blocks() as demo:
                         minimum=1,
                         maximum=50,
                         step=1,
-                        value=20,
+                        value=8,
                     )
             
             with gr.Accordion("R2 Settings", open=False):