Update app.py

app.py

@@ -30,10 +30,10 @@ else:
 huggingface_token = os.getenv("HUGGINFACE_TOKEN")
 
 model_path = snapshot_download(
-    repo_id="black-forest-labs/FLUX.1-dev", 
+    repo_id="LPX55/FLUX.1-merged_uncensored", 
     repo_type="model", 
     ignore_patterns=["*.md", "*..gitattributes"],
-    local_dir="FLUX.1-dev",
+    local_dir="FLUX.1-merged_uncensored",
     token=huggingface_token, # type a new token-id.
 )
 
@@ -45,10 +45,10 @@ controlnet = FluxControlNetModel.from_pretrained(
 pipe = FluxControlNetPipeline.from_pretrained(
     model_path, controlnet=controlnet, torch_dtype=torch.bfloat16
 )
-pipe.load_lora_weights(
-    hf_hub_download("ByteDance/Hyper-SD", "Hyper-FLUX.1-dev-8steps-lora.safetensors"), adapter_name="hyper-sd"
-)
-pipe.set_adapters(["hyper-sd"], adapter_weights=[0.125])
+# pipe.load_lora_weights(
+#     hf_hub_download("ByteDance/Hyper-SD", "Hyper-FLUX.1-dev-8steps-lora.safetensors"), adapter_name="hyper-sd"
+# )
+# pipe.set_adapters(["hyper-sd"], adapter_weights=[0.125])
 pipe.to(device)
 MAX_SEED = 1000000
 MAX_PIXEL_BUDGET = 1024 * 1024