Update app.py

app.py

@@ -14,8 +14,8 @@ from kolors.models.tokenization_chatglm import ChatGLMTokenizer
 from kolors.models.unet_2d_condition import UNet2DConditionModel
 from diffusers import AutoencoderKL, EulerDiscreteScheduler
 
-from huggingface_hub import snapshot_download
 
+from huggingface_hub import hf_hub_download, model_info
 
 device = "cuda"
 root_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))