Update app.py

app.py

@@ -21,7 +21,7 @@ def process(input_image, prompt, low_threshold, high_threshold):
     
     # Convert to tensor and process with model
     with torch.no_grad():
-        c_t = transforms.ToTensor()(canny).unsqueeze(0).cuda()
+        c_t = transforms.ToTensor()(canny).unsqueeze(0)
         output_image = model(c_t, prompt)
         output_pil = transforms.ToPILImage()(output_image[0].cpu() * 0.5 + 0.5)