Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -9,6 +9,8 @@ def get_completion(prompt,negative_prompt,steps,guidance,width,height):
|
|
9 |
def generate(prompt,negative_prompt,steps,guidance,width,height):
|
10 |
output = get_completion(prompt=prompt, height=height, width=width, steps=steps, guidance=guidance, negative_prompt=negative_prompt)
|
11 |
return output
|
|
|
|
|
12 |
|
13 |
with gr.Blocks() as demo:
|
14 |
gr.Markdown("# Image Generation with Stable Diffusion")
|
|
|
9 |
def generate(prompt,negative_prompt,steps,guidance,width,height):
|
10 |
output = get_completion(prompt=prompt, height=height, width=width, steps=steps, guidance=guidance, negative_prompt=negative_prompt)
|
11 |
return output
|
12 |
+
|
13 |
+
pipeline = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16)
|
14 |
|
15 |
with gr.Blocks() as demo:
|
16 |
gr.Markdown("# Image Generation with Stable Diffusion")
|