app

app.py

@@ -23,11 +23,16 @@ from transformers import BlipProcessor, BlipForConditionalGeneration
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 import os
-os.system("git lfs install")
-os.system("git clone https://huggingface.co/h94/IP-Adapter")
-os.system("mv IP-Adapter/sdxl_models sdxl_models")
-os.system('rm -f IP-Adapter/models')
-base_model_path =  "stabilityai/stable-diffusion-xl-base-1.0"
+# os.system("git lfs install")
+# os.system("git clone https://huggingface.co/h94/IP-Adapter")
+# os.system("mv IP-Adapter/sdxl_models sdxl_models")
+
+from huggingface_hub import hf_hub_download
+
+hf_hub_download(repo_id="h94/IP-Adapter", filename="sdxl_models/image_encoder", local_dir="./sdxl_models/image_encoder")
+hf_hub_download(repo_id="InstantX/CSGO", filename="csgo_4_32.bin", local_dir="./CSGO/csgo_4_32.bin")
+# os.system('rm -rf IP-Adapter/models')
+base_model_path = "stabilityai/stable-diffusion-xl-base-1.0"
 image_encoder_path = "sdxl_models/image_encoder"
 csgo_ckpt ='./CSGO/csgo_4_32.bin'
 pretrained_vae_name_or_path ='madebyollin/sdxl-vae-fp16-fix'
@@ -42,8 +47,8 @@ os.system('rm -rf TTPLanet_SDXL_Controlnet_Tile_Realistic/TTPLANET_Controlnet_Ti
 controlnet_path = "./TTPLanet_SDXL_Controlnet_Tile_Realistic"
 
 
-os.system('git clone https://huggingface.co/InstantX/CSGO')
-os.system('rm -rf CSGO/csgo.bin')
+# os.system('git clone https://huggingface.co/InstantX/CSGO')
+# os.system('rm -rf CSGO/csgo.bin')
 
 
 
@@ -151,7 +156,7 @@ def image_grid(imgs, rows, cols):
     for i, img in enumerate(imgs):
         grid.paste(img, box=(i % cols * w, i // cols * h))
     return grid
-@spaces.GPU(enable_queue=True,duration=200)
+@spaces.GPU(enable_queue=True,duration=500)
 def create_image(content_image_pil,
                  style_image_pil,
                  prompt,