Update gradio_app.py

gradio_app.py

@@ -114,6 +114,8 @@ save_fps = 8
 print("resolution:", resolution)
 print("init done.")
 
+
+
 def transpose_if_needed(tensor):
     h = tensor.shape[-2]
     w = tensor.shape[-1]
@@ -160,6 +162,8 @@ def get_image(image1, prompt, image2, dim_steps=50, ddim_eta=1., fs=None, seed=1
     # _, filename = os.path.split(file_list[idx*2])
 
     global model
+    model.cuda()
+    
     ddim_sampler = DDIMSampler(model) if not multiple_cond_cfg else DDIMSampler_multicond(model)
     batch_size = 1
     fs = torch.tensor([fs], dtype=torch.long, device=model.device)