Spaces:
Runtime error
Runtime error
RamAnanth1
commited on
Commit
•
84cc295
1
Parent(s):
17a578d
Update app.py
Browse files
app.py
CHANGED
@@ -4,8 +4,8 @@ import os
|
|
4 |
|
5 |
pipe = pipeline('text-generation', model='RamAnanth1/distilgpt2-sd-prompts')
|
6 |
|
7 |
-
stable_diffusion = gr.
|
8 |
-
clip_interrogator_2 = gr.
|
9 |
|
10 |
def get_images(prompt):
|
11 |
gallery_dir = stable_diffusion(prompt, fn_index=2)
|
|
|
4 |
|
5 |
pipe = pipeline('text-generation', model='RamAnanth1/distilgpt2-sd-prompts')
|
6 |
|
7 |
+
stable_diffusion = gr.load(name="spaces/runwayml/stable-diffusion-v1-5")
|
8 |
+
clip_interrogator_2 = gr.load(name="spaces/fffiloni/CLIP-Interrogator-2")
|
9 |
|
10 |
def get_images(prompt):
|
11 |
gallery_dir = stable_diffusion(prompt, fn_index=2)
|