Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -1,5 +1,6 @@
|
|
| 1 |
from diffusers import StableDiffusionPipeline
|
| 2 |
import gc
|
|
|
|
| 3 |
|
| 4 |
pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4").to("cpu")
|
| 5 |
text_encoder = pipe.text_encoder
|
|
@@ -465,7 +466,7 @@ result=pv_pipe(text_prompt.value,num_inference_steps=num_steps.value,seed=seed.v
|
|
| 465 |
|
| 466 |
|
| 467 |
@spaces.GPU(enable_queue=True)
|
| 468 |
-
|
| 469 |
|
| 470 |
def generate_text(text,seed,num_steps,strength,_=gr.Progress(track_tqdm=True)):
|
| 471 |
result=pv_pipe(text,num_inference_steps=num_steps,seed=seed)
|
|
|
|
| 1 |
from diffusers import StableDiffusionPipeline
|
| 2 |
import gc
|
| 3 |
+
import gradio as gr
|
| 4 |
|
| 5 |
pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4").to("cpu")
|
| 6 |
text_encoder = pipe.text_encoder
|
|
|
|
| 466 |
|
| 467 |
|
| 468 |
@spaces.GPU(enable_queue=True)
|
| 469 |
+
|
| 470 |
|
| 471 |
def generate_text(text,seed,num_steps,strength,_=gr.Progress(track_tqdm=True)):
|
| 472 |
result=pv_pipe(text,num_inference_steps=num_steps,seed=seed)
|