Hugging Face's logo

Spaces:
KarthikAI
/
Sticker_Diffusion
Running

App Files Community
KarthikAI commited on
Commit
8ecbb74
·
verified ·
1 Parent(s): 41ddd36

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +7 -4
utils.py CHANGED
@@ -41,12 +41,15 @@ pipe.load_ip_adapter(
41
 
42
  # Load vision encoder and processor for IP-Adapter embedding
43
  vision_encoder = CLIPVisionModelWithProjection.from_pretrained(
44
- f"{IPADAPTER_REPO}/clip_vision_model",
45
- cache_dir=CACHE_DIR,
 
46
  )
 
47
  image_processor = CLIPImageProcessor.from_pretrained(
48
- f"{IPADAPTER_REPO}/clip_vision_model",
49
- cache_dir=CACHE_DIR,
 
50
  )
51
 
52
  def generate_sticker(input_image: Image.Image, prompt: str):
 
41
 
42
  # Load vision encoder and processor for IP-Adapter embedding
43
  vision_encoder = CLIPVisionModelWithProjection.from_pretrained(
44
+ "h94/IP-Adapter", # repo_id (main IP-Adapter repo)
45
+ subfolder="clip_vision_model",# subfolder within the repo!
46
+ cache_dir=CACHE_DIR
47
  )
48
+
49
  image_processor = CLIPImageProcessor.from_pretrained(
50
+ "h94/IP-Adapter",
51
+ subfolder="clip_vision_model",
52
+ cache_dir=CACHE_DIR
53
  )
54
 
55
  def generate_sticker(input_image: Image.Image, prompt: str):