Update app.py

app.py

@@ -26,14 +26,15 @@ import ruclip
 
 # prepare models:
 #device = 'cuda'
-device = "cuda" if torch.cuda.is_available() else "cpu"
+device = "cuda" if torch.cuda.is_available() else "CPU"
+print('Using device:', device)
 #fp16=True
 #if device == 'CPU':
 #    fp16 = False
 
 model = get_rudalle_model('Malevich', pretrained=True, fp16=True, device=device)
 tokenizer = get_tokenizer()
-vae = get_vae(dwt=True).to(device)
+vae = get_vae(dwt=True).to('cuda')
 
 # pipeline utils:
 #TEST--------