Update app.py

app.py

@@ -71,7 +71,7 @@ def get_seed(seed):
 @spaces.GPU(duration=60)
 def generate(input=DEFAULT_INPUT, filter_input="", negative_input=DEFAULT_NEGATIVE_INPUT, model=DEFAULT_MODEL, height=DEFAULT_HEIGHT, width=DEFAULT_WIDTH, steps=1, guidance=0, number=1, seed=None):
 
-    repo = repo_customs[model or "Default"].to(DEVICE)
+    repo = repo_customs[model or "Default"]
     filter_input = filter_input or ""
     negative_input = negative_input or DEFAULT_NEGATIVE_INPUT
     steps_set = steps
@@ -101,8 +101,10 @@ def generate(input=DEFAULT_INPUT, filter_input="", negative_input=DEFAULT_NEGATI
         guidance_set = 7
         print("1")
         repo.load_lora_weights("ehristoforu/dalle-3-xl-v2", adapter_name="base")
-        repo.set_adapters(["base"], adapter_weights=[0.7])
         print(2)
+        repo.set_adapters(["base"], adapter_weights=[0.7])
+        print(3)
+        repo.to(DEVICE)
 
     if not steps:
         steps = steps_set