Update app.py

app.py

@@ -3,8 +3,8 @@ import gradio as gr
 import torch
 from torch import autocast
 from kandinsky2 import get_kandinsky2
-
-model = get_kandinsky2('cuda', task_type='text2img')
+device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+model = get_kandinsky2(device, task_type='text2img')