Update app.py

app.py

@@ -24,8 +24,8 @@ from rudalle.utils import seed_everything
 #from ruclip import load as get_ruclip
 
 # prepare models:
-device = 'cuda'
-#device = torch.device("cuda:0" if torch.cuda.is_available() else "CPU")
+#device = 'cuda'
+device = "cuda" if torch.cuda.is_available() else "cpu"
 model = get_rudalle_model('Malevich', pretrained=True, fp16=True, device=device)
 tokenizer = get_tokenizer()
 vae = get_vae(dwt=True).to(device)