Update app.py

app.py

@@ -91,14 +91,15 @@ if not os.path.exists(ip_adapter_weights):
 # Initialize model with empty weights
 with init_empty_weights():
     pipe = StableDiffusionXLPipeline.from_pretrained(
-        "Kwai-Kolors/Kolors-IP-Adapter-FaceID-Plus",
+        "./",  # Use local model path
         torch_dtype=dtype,
         safety_checker=None,
+        local_files_only=True  # Force local file usage
     )
 
 # Load and dispatch model with accelerate
 pipe = load_checkpoint_and_dispatch(pipe, model_path, device_map="cpu", offload_folder=None)
-pipe.load_ip_adapter("Kwai-Kolors/Kolors-IP-Adapter-FaceID-Plus", subfolder=None, weight_name="ipa-faceid-plus.bin")
+pipe.load_ip_adapter("./", subfolder=None, weight_name="ipa-faceid-plus.bin")
 
 def generate_image(uploaded_image, prompt):
     img = cv2.cvtColor(np.array(uploaded_image), cv2.COLOR_RGB2BGR)
@@ -130,4 +131,4 @@ interface = gr.Interface(
     description="Upload an image with a face, enter a prompt, and generate a new image preserving the reference face."
 )
 
-interface.launch()
+interface.launch()