Update app.py

app.py

@@ -4,7 +4,7 @@ from torch import autocast
 from kandinsky2 import get_kandinsky2
 
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-model = get_kandinsky2('cuda', task_type='text2img', cache_dir='/tmp/kandinsky2', model_version='2.1', use_flash_attention=False)
+model = get_kandinsky2(device.type, task_type='text2img', cache_dir='/tmp/kandinsky2', model_version='2.1', use_flash_attention=False)
 
 def generate_text(prompt, quality="High (Default)"):
     length_dict = {"Low": 50, "High (Default)": 100, "Ultra": 150}