Hugging Face's logo

Spaces:
Suchinthana
/
VisualizeGeoMap
Running on Zero

App Files Community
Suchinthana commited on
Commit
1b5be08
·
1 Parent(s): 6b6be99

model update

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -108,7 +108,7 @@ def save_map_image(geojson_data):
108
  # ControlNet pipeline setup
109
  controlnet = ControlNetModel.from_pretrained("lllyasviel/control_v11p_sd15_inpaint", torch_dtype=torch.float16)
110
  pipeline = StableDiffusionControlNetInpaintPipeline.from_pretrained(
111
- "runwayml/stable-diffusion-inpainting", controlnet=controlnet, torch_dtype=torch.float16
112
  )
113
  pipeline.enable_model_cpu_offload()
114
 
 
108
  # ControlNet pipeline setup
109
  controlnet = ControlNetModel.from_pretrained("lllyasviel/control_v11p_sd15_inpaint", torch_dtype=torch.float16)
110
  pipeline = StableDiffusionControlNetInpaintPipeline.from_pretrained(
111
+ "stable-diffusion-v1-5/stable-diffusion-inpainting", controlnet=controlnet, torch_dtype=torch.float16
112
  )
113
  pipeline.enable_model_cpu_offload()
114