Update app.py

app.py

@@ -4,14 +4,13 @@ import torch
 from torch import autocast
 from kandinsky2 import get_kandinsky2
 
-#model_id = "hakurei/waifu-diffusion"
 model = get_kandinsky2('cuda', task_type='text2img')
 
-#torch.backends.cudnn.benchmark = True
+
 
 
 def infer(prompt):
-    images = model.generate_text2img('A teddy bear на красной площади', batch_size=4, h=512, w=512, num_steps=75, denoised_type='dynamic_threshold', dynamic_threshold_v=99.5, sampler='ddim_sampler', ddim_eta=0.05, guidance_scale=10)
+    images = model.generate_text2img(prompt, batch_size=4, h=512, w=512, num_steps=75, denoised_type='dynamic_threshold', dynamic_threshold_v=99.5, sampler='ddim_sampler', ddim_eta=0.05, guidance_scale=10)
     return images
 
 css = """