Update app.py

app.py

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