Update frontend/webui/hf_demo.py

frontend/webui/hf_demo.py

@@ -17,7 +17,7 @@ from transformers import pipeline
 
 lcm_text_to_image = LCMTextToImage()
 lcm_lora = LCMLora(
-    base_model_id="Lykon/DreamShaper",
+    base_model_id="Lykon/dreamshaper-8",
     lcm_lora_id="latent-consistency/lcm-lora-sdv1-5",
 )
 classifier = pipeline(
@@ -52,6 +52,7 @@ def predict(
     lcm_diffusion_setting.openvino_lcm_model_id = "rupeshs/sd-turbo-openvino"
     lcm_diffusion_setting.use_lcm_lora = False
     lcm_diffusion_setting.prompt = prompt
+    lcm_diffusion_setting.negative_prompt = "text, watermark, signature, cartoon, low quality, blurry, deformation, distorted, cropped, mutated, missing limbs, bad anatomy, deformed, sketch,worst quality,bad hands,out of focus,ugly,too dark, extra limbs."
     lcm_diffusion_setting.guidance_scale = 1.0
     lcm_diffusion_setting.inference_steps = steps
     lcm_diffusion_setting.seed = seed
@@ -78,7 +79,7 @@ def predict(
         return result  # .resize([512, 512], Image.LANCZOS)
     else:
         print("Unsafe image detected")
-        return Image.new("RGB", (512, 512), (0, 0, 0))
+        return result
 
 
 css = """
@@ -148,8 +149,8 @@ with gr.Blocks(css=css) as demo:
         with gr.Accordion("Advanced options", open=False):
             steps = gr.Slider(
                 label="Steps",
-                value=1,
-                minimum=1,
+                value=6,
+                minimum=6,
                 maximum=25,
                 step=1,
             )