Update app.py

app.py

@@ -3,10 +3,10 @@ import os; os.system("pip install --pre --upgrade spaces")
 import random
 import uuid
 
-import spaces
 import gradio as gr
 import numpy as np
 from PIL import Image
+import spaces
 import torch
 from diffusers import StableDiffusionXLPipeline, EulerAncestralDiscreteScheduler
 
@@ -40,12 +40,14 @@ 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")