Update app.py

app.py

@@ -743,6 +743,9 @@ def generate_image(
     device = "cuda" if torch.cuda.is_available() else "cpu"
     torch_device = torch.device(device)
 
+    global model
+    model = model.to(torch_device)
+
     if do_img2img and init_image is not None:
         init_image = get_image(init_image)
         if resize_img: