Update app.py

app.py

@@ -26,7 +26,11 @@ import ruclip
 
 # prepare models:
 #device = 'cuda'
-device = "cuda" if torch.cuda.is_available() else "cpu"
+device1 = "cuda:0" if torch.cuda.is_available() else "CPU"
+if device1 = 'CPU':
+    device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+raise ValueError([device1, device])
+
 model = get_rudalle_model('Malevich', pretrained=True, fp16=True, device=device)
 tokenizer = get_tokenizer()
 vae = get_vae(dwt=True).to(device)