Update app.py

app.py

@@ -54,8 +54,10 @@ def process(input_image, prompt, a_prompt, n_prompt, num_samples, image_resoluti
         detected_map = cv2.GaussianBlur(detected_map, (0, 0), 3.0)
         detected_map[detected_map > 4] = 255
         detected_map[detected_map < 255] = 0
-
-        control = torch.from_numpy(detected_map.copy()).float().cuda() / 255.0
+        if torch.cuda.is_available():
+            control = torch.from_numpy(detected_map.copy()).float().cuda() / 255.0
+        else:
+            control = torch.from_numpy(detected_map.copy()).float() / 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()