Update main_code_script.py

main_code_script.py

@@ -60,7 +60,9 @@ def inpaint_clothing(image, mask_img, clothing_prompt, device="cuda" if torch.cu
     using Stable Diffusion Inpainting.
     """
     pipe = StableDiffusionInpaintPipeline.from_pretrained(
-        "stabilityai/stable-diffusion-2-inpainting",
+#        "stabilityai/stable-diffusion-2-inpainting",
+        "stabilityai/stable-diffusion-2-base"
+#        "SimianLuo/LCM_Dreamshaper_v7"
         torch_dtype=torch.float16
     )
     pipe = pipe.to(device)