Update app.py

app.py

@@ -28,7 +28,7 @@ from dataclasses import dataclass
 import math
 from pathlib import Path
 from typing import Any, Callable, Dict, List, Optional, Union
-from huggingface_hub import hf_hub_download
+from huggingface_hub import hf_hub_download, snapshot_download
 from diffusers import DiffusionPipeline, AutoencoderTiny, AutoPipelineForImage2Image, FluxTransformer2DModel
 import safetensors.torch
 from safetensors.torch import load_file