pharmapsychotic commited on
Commit
101d8cd
1 Parent(s): 55bc038

Links for ViT-L and ViT-H models

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -26,12 +26,12 @@ for file in PREPROCESS_FILES:
26
  cache_path = os.path.dirname(path)
27
 
28
 
29
- # load BLIP and ViT-L
30
  config = Config(cache_path=cache_path, clip_model_path="cache", clip_model_name="ViT-L-14/openai")
31
  ci_vitl = Interrogator(config)
32
  ci_vitl.clip_model = ci_vitl.clip_model.to("cpu")
33
 
34
- # load ViT-H
35
  config.blip_model = ci_vitl.blip_model
36
  config.clip_model_name = "ViT-H-14/laion2b_s32b_b79k"
37
  ci_vith = Interrogator(config)
@@ -80,7 +80,7 @@ TITLE = """
80
  </h1>
81
  </div>
82
  <p style="margin-bottom: 10px; font-size: 94%">
83
- Want to figure out what a good prompt might be to create new images like an existing one? The CLIP Interrogator is here to get you answers!
84
  </p>
85
  </div>
86
  """
@@ -99,7 +99,7 @@ ARTICLE = """
99
 
100
  <p>
101
  Has this been helpful to you? Follow me on twitter
102
- <a href="https://twitter.com/pharmapsychotic">@pharmapsychotic</a>
103
  and check out more tools at my
104
  <a href="https://pharmapsychotic.com/tools.html">Ai generative art tools list</a>
105
  </p>
 
26
  cache_path = os.path.dirname(path)
27
 
28
 
29
+ # load BLIP and ViT-L https://huggingface.co/openai/clip-vit-large-patch14
30
  config = Config(cache_path=cache_path, clip_model_path="cache", clip_model_name="ViT-L-14/openai")
31
  ci_vitl = Interrogator(config)
32
  ci_vitl.clip_model = ci_vitl.clip_model.to("cpu")
33
 
34
+ # load ViT-H https://huggingface.co/laion/CLIP-ViT-H-14-laion2B-s32B-b79K
35
  config.blip_model = ci_vitl.blip_model
36
  config.clip_model_name = "ViT-H-14/laion2b_s32b_b79k"
37
  ci_vith = Interrogator(config)
 
80
  </h1>
81
  </div>
82
  <p style="margin-bottom: 10px; font-size: 94%">
83
+ Want to figure out what a good prompt might be to create new images like an existing one?<br>The CLIP Interrogator is here to get you answers!
84
  </p>
85
  </div>
86
  """
 
99
 
100
  <p>
101
  Has this been helpful to you? Follow me on twitter
102
+ <a href="https://twitter.com/pharmapsychotic">@pharmapsychotic</a><br>
103
  and check out more tools at my
104
  <a href="https://pharmapsychotic.com/tools.html">Ai generative art tools list</a>
105
  </p>