Spaces:
Runtime error
Runtime error
File size: 618 Bytes
0b00c74 d2794b1 0b00c74 ed6eb7f d2794b1 0b00c74 ed6eb7f 0b00c74 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
import gradio as gr
from optimum.intel.openvino import OVStableDiffusionPipeline
pipe = OVStableDiffusionPipeline.from_pretrained("OpenVINO/stable-diffusion-2-1-quantized", compile=False)
pipe.reshape(batch_size=1, height=512, width=512, num_images_per_prompt=1)
pipe.compile()
prompt = "sailing ship in storm by Rembrandt"
def generate(name):
output = pipe(prompt, num_inference_steps=50, output_type="pil")
output.images[0].save("result.png")
return output.images[0]
iface = gr.Interface(fn=generate, inputs=[gr.Textbox(label="Prompt", value=prompt)], outputs=[gr.Image(type="pil")])
iface.launch()
|