Update app.py
Browse files
app.py
CHANGED
@@ -11,8 +11,8 @@ Stable Diffusion 2 uses OpenCLIP ViT-H model trained on LAION dataset so it know
|
|
11 |
</p>
|
12 |
"""
|
13 |
|
14 |
-
stable_diffusion = gr.
|
15 |
-
clip_interrogator_2 = gr.
|
16 |
|
17 |
def get_images(prompt):
|
18 |
gallery_dir = stable_diffusion(prompt, fn_index=2)
|
|
|
11 |
</p>
|
12 |
"""
|
13 |
|
14 |
+
stable_diffusion = gr.load(name="spaces/runwayml/stable-diffusion-v1-5")
|
15 |
+
clip_interrogator_2 = gr.load(name="spaces/fffiloni/CLIP-Interrogator-2")
|
16 |
|
17 |
def get_images(prompt):
|
18 |
gallery_dir = stable_diffusion(prompt, fn_index=2)
|