Update frontend/webui/hf_demo.py

frontend/webui/hf_demo.py

@@ -60,8 +60,8 @@ def predict(
     lcm_diffusion_setting.use_tiny_auto_encoder = False
     # lcm_diffusion_setting.image_width = 320 if is_openvino_device() else 512
     # lcm_diffusion_setting.image_height = 320 if is_openvino_device() else 512
-    lcm_diffusion_setting.image_width = 724
-    lcm_diffusion_setting.image_height = 724
+    lcm_diffusion_setting.image_width = 768
+    lcm_diffusion_setting.image_height = 768
     lcm_diffusion_setting.use_openvino = True
     lcm_diffusion_setting.use_tiny_auto_encoder = True
     pprint(lcm_diffusion_setting.model_dump())
@@ -78,7 +78,7 @@ def predict(
         return result  # .resize([512, 512], Image.LANCZOS)
     else:
         print("Unsafe image detected")
-        return Image.new("RGB", (724, 724), (0, 0, 0))
+        return Image.new("RGB", (768, 768), (0, 0, 0))
 
 
 css = """