Update app.py

app.py

@@ -1,3 +1,4 @@
+import spaces
 import subprocess
 subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
 
@@ -24,7 +25,6 @@ from torchvision.transforms.functional import to_tensor
 from transformers import AutoTokenizer, T5EncoderModel
 from huggingface_hub import hf_hub_download
 import gradio as gr
-import spaces
 
 from models.infinity import Infinity
 from models.basic import *