Update app.py

app.py

@@ -117,6 +117,12 @@ def process(
     height, width = control.size(-2), control.size(-1)
     model.control_scales = [strength] * 13
 
+    # Move model and tensors to GPU if available
+    if torch.cuda.is_available():
+        model.to("cuda")
+        control = control.to("cuda")
+        x_T = x_T.to("cuda")
+
     sampler = SpacedSampler(model, var_type="fixed_small")
     preds = []
     for _ in tqdm(range(num_samples)):