Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -12,10 +12,11 @@ Stable Diffusion 2 uses OpenCLIP ViT-H model trained on LAION dataset so it know
|
|
12 |
</p>
|
13 |
"""
|
14 |
|
15 |
-
stable_diffusion =
|
16 |
clip_interrogator_2 = Client("https://fffiloni-clip-interrogator-2.hf.space/")
|
17 |
|
18 |
def get_images(prompt):
|
|
|
19 |
gallery_dir = stable_diffusion(prompt, fn_index=2)
|
20 |
img_results = [os.path.join(gallery_dir, img) for img in os.listdir(gallery_dir)]
|
21 |
return img_results[0]
|
|
|
12 |
</p>
|
13 |
"""
|
14 |
|
15 |
+
stable_diffusion = Client("https://runwayml-stable-diffusion-v1-5.hf.space/")
|
16 |
clip_interrogator_2 = Client("https://fffiloni-clip-interrogator-2.hf.space/")
|
17 |
|
18 |
def get_images(prompt):
|
19 |
+
print("Calling SD")
|
20 |
gallery_dir = stable_diffusion(prompt, fn_index=2)
|
21 |
img_results = [os.path.join(gallery_dir, img) for img in os.listdir(gallery_dir)]
|
22 |
return img_results[0]
|