update app.py

app.py

@@ -39,15 +39,16 @@ ckpt = torch.load(model_path, map_location='cpu')
 opts = ckpt['opts']
 opts['checkpoint_path'] = model_path
 opts['pretrained_e4e_path'] = e4e_path
-opts['device'] = 'cuda' if torch.cuda.is_available() else 'cpu'
+device = 'cuda' if torch.cuda.is_available() else 'cpu'
+opts['device'] = device
 opts = Namespace(**opts)
 encoder = pSp(opts)
 encoder.eval()
-encoder.cuda()
+encoder.to(device)
 
 adapter = CLIPAdapterWithDecoder(opts)
 adapter.eval()
-adapter.cuda()
+adapter.to(device)
 
 clip_model, _ = clip.load("ViT-B/32", device='cuda')
 
@@ -56,8 +57,8 @@ def manipulate(input_image, caption):
     input_image = input_transforms(aligned_image)
     input_image = input_image.unsqueeze(0)
     text_input = clip.tokenize(caption)
-    text_input = text_input.cuda()
-    input_image = input_image.cuda().float()
+    text_input = text_input.to(device)
+    input_image = input_image.to(device).float()
 
     with torch.no_grad():
         text_features = clip_model.encode_text(text_input).float()