AlexKoff88 commited on
Commit
86ea458
·
1 Parent(s): aad0875

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -3,7 +3,7 @@ from optimum.intel.openvino import OVStableDiffusionPipeline
3
  from diffusers.training_utils import set_seed
4
 
5
  quantized_pipe = OVStableDiffusionPipeline.from_pretrained("OpenVINO/Stable-Diffusion-Pokemon-en-quantized", compile=False)
6
- quantized_pipe.reshape(batch_size=1, height=256, width=256, num_images_per_prompt=1)
7
  quantized_pipe.compile()
8
 
9
  prompt = "cartoon bird"
@@ -24,5 +24,6 @@ gr.Interface(
24
  outputs=gr.outputs.Image(type="pil", label="Generated Image"),
25
  title="OpenVINO-optimized Stable Diffusion",
26
  description="",
 
27
  theme="huggingface",
28
  ).launch()
 
3
  from diffusers.training_utils import set_seed
4
 
5
  quantized_pipe = OVStableDiffusionPipeline.from_pretrained("OpenVINO/Stable-Diffusion-Pokemon-en-quantized", compile=False)
6
+ quantized_pipe.reshape(batch_size=1, height=512, width=512, num_images_per_prompt=1)
7
  quantized_pipe.compile()
8
 
9
  prompt = "cartoon bird"
 
24
  outputs=gr.outputs.Image(type="pil", label="Generated Image"),
25
  title="OpenVINO-optimized Stable Diffusion",
26
  description="",
27
+ examples=examples,
28
  theme="huggingface",
29
  ).launch()