Update app.py

app.py

@@ -9,6 +9,10 @@ import openvino.runtime as ov
 from typing import Optional, Dict
 
 model_id = "Disty0/LCM_SoteMix"
+HIGH=1024
+WIDTH=1024
+
+
 batch_size = -1
 class CustomOVModelVaeDecoder(OVModelVaeDecoder):
     def __init__(
@@ -22,19 +26,19 @@ pipe = OVStableDiffusionPipeline.from_pretrained(model_id, compile = False, ov_c
 taesd_dir = snapshot_download(repo_id="deinferno/taesd-openvino")
 pipe.vae_decoder = CustomOVModelVaeDecoder(model = OVBaseModel.load_model(f"{taesd_dir}/vae_decoder/openvino_model.xml"), parent_model = pipe, model_dir = taesd_dir)
 
-pipe.reshape( batch_size=-1, height=512, width=512, num_images_per_prompt=1)
+pipe.reshape( batch_size=-1, height=HIGH, width=WIDTH, num_images_per_prompt=1)
 pipe.compile()
 
 prompt=""
 negative_prompt=""
 
-def infer(prompt,negative_prompt):
+def infer(prompt,negative_prompt,step):
 
     image = pipe(
         prompt = prompt, 
         negative_prompt = negative_prompt,
-        width = 512, 
-        height = 512,
+        width = WIDTH, 
+        height = WIDTH,
         guidance_scale=1.0,
         num_inference_steps=8,
         num_images_per_prompt=1,
@@ -63,7 +67,7 @@ with gr.Blocks(css=css) as demo:
     
     with gr.Column(elem_id="col-container"):
         gr.Markdown(f"""
-        # Disty0/LCM_SoteMix 512x512
+        # Disty0/LCM_SoteMix 1024x1024
         Currently running on {power_device}.
         """)