Update demo_gradio.py

demo_gradio.py

@@ -38,7 +38,7 @@ from DPT.dpt.transforms import Resize, NormalizeImage, PrepareForNet
 """
 Get ZeST Ready
 """
-base_model_path = "stabilityai/stable-diffusion-xl-base-1.0"
+base_model_path = "Lykon/dreamshaper-xl-lightning"
 image_encoder_path = "models/image_encoder"
 ip_ckpt = "sdxl_models/ip-adapter_sdxl_vit-h.bin"
 controlnet_path = "diffusers/controlnet-depth-sdxl-1.0"
@@ -161,7 +161,7 @@ def greet(input_image, material_exemplar):
     
     
     num_samples = 1
-    images = ip_model.generate(pil_image=ip_image, image=init_img, control_image=depth_map, mask_image=mask, controlnet_conditioning_scale=0.9, num_samples=num_samples, num_inference_steps=30, seed=42)
+    images = ip_model.generate(guidance_scale=2, pil_image=ip_image, image=init_img, control_image=depth_map, mask_image=mask, controlnet_conditioning_scale=0.9, num_samples=num_samples, num_inference_steps=6, seed=42)
     
     return images[0]