Update app.py

app.py

@@ -44,7 +44,7 @@ def process_canny(input_image, prompt, a_prompt, n_prompt, num_samples, image_re
         detected_map = apply_canny(img, low_threshold, high_threshold)
         detected_map = HWC3(detected_map)
 
-        control = torch.from_numpy(detected_map.copy()).float() / 255.0
+        control = torch.from_numpy(detected_map.copy()).float().cuda() / 255.0
         control = torch.stack([control for _ in range(num_samples)], dim=0)
         control = einops.rearrange(control, 'b h w c -> b c h w').clone()