Upload folder using huggingface_hub

README.md

@@ -7,4 +7,4 @@ tags:
 inference: true
 ---
 
-# Inference Endpoint for [ControlNet](https://huggingface.co/lllyasviel/ControlNet) using [runwayml/stable-diffusion-v1-5](https://huggingface.co/runwayml/stable-diffusion-v1-5)
+# Inference Endpoint for [Seg2Sat](https://huggingface.co/rgres/Seg2Sat-sd-controlnet) using [runwayml/stable-diffusion-v1-5](https://huggingface.co/stabilityai/stable-diffusion-2-1-base)

handler.py

@@ -17,7 +17,7 @@ class EndpointHandler():
     def __init__(self, path=""):
         self.stable_diffusion_id = "stabilityai/stable-diffusion-2-1-base"
 
-        controlnet = ControlNetModel.from_pretrained("rgres/sd-controlnet-aerialdreams", torch_dtype=torch.float16)
+        controlnet = ControlNetModel.from_pretrained("rgres/Seg2Sat-sd-controlnet", torch_dtype=torch.float16)
 
         self.pipe = StableDiffusionControlNetPipeline.from_pretrained(
             self.stable_diffusion_id, controlnet=controlnet, torch_dtype=dtype, safety_checker=None