Update app.py

app.py

@@ -8,7 +8,12 @@ import torch
 def image_mod(prompt: str, model: str, image_0: gr.Image, image_1: gr.Image) -> list[gr.Image]:
     images = [image_0, image_1]
     for i, diffusion_model in enumerate(model):
-        pipeline = DiffusionPipeline.from_pretrained(pretrained_model_name_or_path=diffusion_model, torch_dtype=torch.float16, device_map="auto")
+        pipeline = DiffusionPipeline.from_pretrained(
+            pretrained_model_name_or_path=diffusion_model,
+            torch_dtype=torch.float16,
+            use_safetensors=True,
+            device_map="auto"
+        )
         images[i] = pipeline(prompt).images[0]
     return images