Update gradio_seesr.py

gradio_seesr.py

@@ -14,6 +14,7 @@ from diffusers import AutoencoderKL, DDPMScheduler
 from diffusers.utils import check_min_version
 from diffusers.utils.import_utils import is_xformers_available
 from transformers import CLIPTextModel, CLIPTokenizer, CLIPImageProcessor
+from huggingface_hub import hf_hub_download, snapshot_download
 
 from pipelines.pipeline_seesr import StableDiffusionControlNetPipeline
 
@@ -34,6 +35,17 @@ ram_transforms = transforms.Compose([
             transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
         ])
 
+snapshot_download(
+    repo_id="CSWRY/SeeSR", 
+    local_dir="preset/models"
+)
+
+
+snapshot_download(
+    repo_id="stabilityai/stable-diffusion-2-1-base", 
+    local_dir="preset/models/sd-turbo"
+)
+
 
 # Load scheduler, tokenizer and models.
 pretrained_model_path = 'preset/models/stable-diffusion-2-1-base'