Update app.py

app.py

@@ -33,8 +33,11 @@ save_memory = False
 def load_model():
     model_path = hf_hub_download('lllyasviel/ControlNet', 'models/control_sd15_scribble.pth')
     model = create_model('./models/cldm_v15.yaml').cpu()
-    model.load_state_dict(load_state_dict(model_path, location='cuda'))
-    model = model.cuda()
+    if torch.cuda.is_available():
+        model.load_state_dict(load_state_dict(model_path, location='cuda'))
+        model = model.cuda()
+    else:
+        model.load_state_dict(load_state_dict(model_path, location='cpu'))
     return model
 
 def process(input_image, prompt, a_prompt, n_prompt, num_samples, image_resolution, detect_resolution, ddim_steps, guess_mode, strength, scale, seed, eta):