Update app.py

app.py

@@ -32,9 +32,9 @@ print('Using device:', device)
 #if device == 'CPU':
 #    fp16 = False
 
-model = get_rudalle_model('Malevich', pretrained=True, fp16=True, device=device)
+dalle = get_rudalle_model('Malevich', pretrained=True, fp16=True, device=device)
 tokenizer = get_tokenizer()
-vae = get_vae(dwt=True).to('cuda')
+vae = get_vae(dwt=True).to(device)
 
 # pipeline utils:
 #TEST--------