update model id

handler.py

@@ -6,8 +6,8 @@ from io import BytesIO
 import base64
 
 # Load the model once (caching for efficiency)
-MODEL_ID = "alimama-creative/FLUX.1-dev-Controlnet-Inpainting-Alpha"
-CONTROLNET_MODEL = "alimama-creative/FLUX.1-dev-Controlnet-Inpainting-Alpha"
+MODEL_ID = "jiteshdhamaniya/alimama-creative-FLUX.1-dev-Controlnet-Inpainting-Alpha"
+CONTROLNET_MODEL = "jiteshdhamaniya/alimama-creative-FLUX.1-dev-Controlnet-Inpainting-Alpha"
 TRANSFORMER_MODEL = "black-forest-labs/FLUX.1-dev"
 
 controlnet = DiffusionPipeline.from_pretrained(CONTROLNET_MODEL, torch_dtype=torch.bfloat16)

main.py

@@ -21,7 +21,7 @@ prompt='a person wearing a white shoe, carrying a white bucket with text "Jitesh
 # print(f"Checkpoint file: {checkpoint_file}")
 
 # Build pipeline
-controlnet = FluxControlNetModel.from_pretrained("alimama-creative/FLUX.1-dev-Controlnet-Inpainting-Alpha", torch_dtype=torch.bfloat16, local_files_only=True)
+controlnet = FluxControlNetModel.from_pretrained("jiteshdhamaniya/alimama-creative-FLUX.1-dev-Controlnet-Inpainting-Alpha", torch_dtype=torch.bfloat16, local_files_only=True)
 transformer = FluxTransformer2DModel.from_pretrained(
         "black-forest-labs/FLUX.1-dev", subfolder='transformer', torch_dtype=torch.bfloat16, local_files_only=True
     )