Update app.py

app.py

@@ -5,13 +5,13 @@ from huggingface_hub import snapshot_download
 import openvino.runtime as ov
 from typing import Optional, Dict
 
-model_id = "hsuwill000/Fluently-v4-LCM-openvino"
+model_id = "rupeshs/sdxs-512-0.9-openvino"
 
 HIGH = 1024
 WIDTH = 512
 
 batch_size = -1  # Or set it to a specific positive integer if needed
-
+"""
 class CustomOVModelVaeDecoder(OVModelVaeDecoder):
     def __init__(
         self, model: ov.Model, parent_model: OVBaseModel, ov_config: Optional[Dict[str, str]] = None, model_dir: str = None,
@@ -19,6 +19,8 @@ class CustomOVModelVaeDecoder(OVModelVaeDecoder):
         super(OVModelVaeDecoder, self).__init__(model, parent_model, ov_config, "vae_decoder", model_dir)
 
 
+"""
+
 
 pipe = OVStableDiffusionPipeline.from_pretrained(
     model_id,
@@ -28,13 +30,15 @@ pipe = OVStableDiffusionPipeline.from_pretrained(
     safety_checker=None,
     use_safetensors=False,
 )
-
+"""
 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
-                                          )
+       
+"""
+                                   )
 
 print(pipe.scheduler.compatibles)