Spaces:
Runtime error
Runtime error
pharmapsychotic
commited on
Commit
•
101d8cd
1
Parent(s):
55bc038
Links for ViT-L and ViT-H models
Browse files
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
|
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>
|