Update app.py

app.py

@@ -30,7 +30,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 #device = "cuda"
 print('Using device:', device)
 
-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(device)
 
@@ -54,8 +54,8 @@ text = 'радуга на фоне ночного города'
 def model(text, tokenizer=tokenizer, dalle=dalle, vae=vae, top_k=2048, images_num=1, bs=8, top_p=0.995):
     # _pil_images, _scores = generate_images(text, tokenizer, dalle, vae, top_k=top_k, images_num=images_num, bs=8, top_p=top_p)
     pil_images = generate_images(text, tokenizer, dalle, vae, top_k=top_k, images_num=images_num, bs=8, top_p=top_p)
-    pil_images = pil_images[0]
-    return pil_images
+#    pil_images = pil_images[0]
+    return pil_images[0]
 
 seed_everything(42)
 pil_images = []