Update app.py

app.py

@@ -43,7 +43,7 @@ controlnet = FluxControlNetModel.from_pretrained(
     "jasperai/Flux.1-dev-Controlnet-Upscaler", torch_dtype=torch.bfloat16
 ).to(device)
 pipe = FluxControlNetPipeline.from_pretrained(
-    model_path, controlnet=controlnet, torch_dtype=torch.bfloat16
+    "LPX55/FLUX.1-merged_uncensored", controlnet=controlnet, torch_dtype=torch.bfloat16, token=huggingface_token,
 )
 # pipe.load_lora_weights(
 #     hf_hub_download("ByteDance/Hyper-SD", "Hyper-FLUX.1-dev-8steps-lora.safetensors"), adapter_name="hyper-sd"
@@ -51,7 +51,7 @@ pipe = FluxControlNetPipeline.from_pretrained(
 # pipe.set_adapters(["hyper-sd"], adapter_weights=[0.125])
 pipe.to(device)
 MAX_SEED = 1000000
-MAX_PIXEL_BUDGET = 1024 * 1024
+MAX_PIXEL_BUDGET = 2048 * 2048
 
 
 def process_input(input_image, upscale_factor, **kwargs):