Update app.py

app.py

@@ -43,16 +43,16 @@ footer {
 
 repo_nsfw_classifier = pipeline("image-classification", model="Falconsai/nsfw_image_detection")
 
-repo_default = DiffusionPipeline.from_pretrained("fluently/Fluently-XL-Final", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False).to(DEVICE)
+repo_default = DiffusionPipeline.from_pretrained("fluently/Fluently-XL-Final", device=DEVICE, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False).to(DEVICE)
 
-repo_neo = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16, use_safetensors=True, add_watermarker=False, revision="refs/pr/1").to(DEVICE)
+repo_neo = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", device=DEVICE, torch_dtype=torch.bfloat16, use_safetensors=True, add_watermarker=False, revision="refs/pr/1").to(DEVICE)
 repo_neo.load_lora_weights("ehristoforu/dalle-3-xl-v2", adapter_name="base")
 repo_neo.set_adapters(["base"], adapter_weights=[0.7])
 
 repo_customs = {
     "Default": repo_default,
-    "Realistic": DiffusionPipeline.from_pretrained("ehristoforu/Visionix-alpha", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False).to(DEVICE),
-    "Anime": DiffusionPipeline.from_pretrained("cagliostrolab/animagine-xl-3.1", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False).to(DEVICE),
+    "Realistic": DiffusionPipeline.from_pretrained("ehristoforu/Visionix-alpha", device=DEVICE, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False).to(DEVICE),
+    "Anime": DiffusionPipeline.from_pretrained("cagliostrolab/animagine-xl-3.1", device=DEVICE, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False).to(DEVICE),
     "Pixel": repo_default,
     "Neo": repo_neo,
 }