Update app.py

app.py

@@ -1,6 +1,9 @@
 import gradio as gr
+import torch
+from torch import autocast
 from kandinsky2 import get_kandinsky2
-model = get_kandinsky2('cpu', task_type='text2img', cache_dir='/tmp/kandinsky2', model_version='2.1', use_flash_attention=False)
+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)
 def generate_text(prompt, quality="High (Default)"):
     length_dict = {"Low": 50, "High (Default)": 100, "Ultra": 150}
     length = length_dict[quality]