Update app.py

app.py

@@ -108,7 +108,7 @@ def infer(
     width=512, 
     height=512, 
     num_inference_steps=50, 
-    model='stable-diffusion-v1-5/stable-diffusion-v1-5', 
+    model=model_default, #'stable-diffusion-v1-5/stable-diffusion-v1-5', 
     seed=4, 
     guidance_scale=7.5, 
     lora_scale=0.7,
@@ -299,10 +299,8 @@ def infer(
                     generator = torch.Generator(device).manual_seed(seed)
 
                     # Преобразование изображений для IP-Adapter (размер 224x224)
-                    # ip_source_image = preprocess_image(ip_source_image, width, height, resize_to_224=True)
-                    # ip_adapter_image = preprocess_image(ip_adapter_image, width, height, resize_to_224=True)
-                    ip_source_image = preprocess_image(ip_source_image, width, height)
-                    ip_adapter_image = preprocess_image(ip_adapter_image, width, height)
+                    ip_source_image = preprocess_image(ip_source_image, width, height, resize_to_224=True)
+                    ip_adapter_image = preprocess_image(ip_adapter_image, width, height, resize_to_224=True)
 
                     pipe_ip_adapter = StableDiffusionControlNetPipeline.from_pretrained(
                             model_default,
@@ -405,7 +403,7 @@ def infer(
                     generator = torch.Generator(device).manual_seed(seed)
 
                     pipe_controlnet = StableDiffusionControlNetPipeline.from_pretrained(
-                        "stable-diffusion-v1-5/stable-diffusion-v1-5", 
+                        model_default, #"stable-diffusion-v1-5/stable-diffusion-v1-5", 
                         controlnet=controlnet, 
                         torch_dtype=torch_dtype, 
                         use_safetensors=True
@@ -445,7 +443,7 @@ def infer(
                         generator = torch.Generator(device).manual_seed(seed)
 
                         pipe_controlnet = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
-                            "stable-diffusion-v1-5/stable-diffusion-v1-5",
+                            model_default, #"stable-diffusion-v1-5/stable-diffusion-v1-5",
                             controlnet=controlnet,
                             torch_dtype=torch_dtype,
                             use_safetensors=True