fix more

app.py

@@ -4,7 +4,7 @@ import gradio as gr
 from datasets import load_dataset
 from PIL import Image
 
-from model import get_sd_small
+from model import get_sd_small, get_sd_tiny
 from trans_google import google_translator
 
 from i18n import i18nTranslator
@@ -43,7 +43,7 @@ samplers = [
 rand = random.Random()
 translator = google_translator()
 
-pipe = get_sd_small()
+pipe = get_sd_tiny()
 
 
 def infer(prompt: str, negative: str, width: int, height: int, sampler: str, steps: int, seed: int, scale):
@@ -133,7 +133,7 @@ css = """
             padding-top: 1.5rem;
         }
         #prompt-column {
-            min-height: 450px
+            min-height: 500px
         }
         #gallery {
             min-height: 22rem;

model.py

@@ -3,6 +3,7 @@ from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler, DPMSolver
     OnnxStableDiffusionPipeline
 
 import pipeline_openvino_stable_diffusion
+from optimum.intel.openvino import OVStableDiffusionPipeline
 
 
 def get_sd_21():
@@ -39,3 +40,10 @@ def get_sd_small():
     )
     pipe = pipeline_openvino_stable_diffusion.OpenVINOStableDiffusionPipeline.from_onnx_pipeline(onnx_pipe)
     return pipe
+
+
+def get_sd_tiny():
+    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()
+    return pipe

requirements.txt

@@ -8,4 +8,5 @@ onnx
 openvino
 onnxruntime-openvino
 ftfy
-py-cpuinfo
+py-cpuinfo
+optimum-intel[openvino, diffusers]