Update app.py

app.py

@@ -40,14 +40,12 @@ if torch.cuda.is_available():
         "fluently/Fluently-XL-Final",
         torch_dtype=torch.float16,
         use_safetensors=True,
-    )
+    ).to("cuda")
     pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
     
     
     pipe.load_lora_weights("ehristoforu/dalle-3-xl-v2", weight_name="dalle-3-xl-lora-v2.safetensors", adapter_name="dalle")
     pipe.set_adapters("dalle")
-
-    pipe.to("cuda")