Update app.py

app.py

@@ -3,11 +3,12 @@ import os
 
 from huggingface_hub import list_liked_repos
 from diffusers import DiffusionPipeline
+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, device_map="auto")
+        pipeline = DiffusionPipeline.from_pretrained(pretrained_model_name_or_path=diffusion_model, torch_dtype=torch.float16, device_map="auto")
         images[i] = pipeline(prompt).images[0]
     return images