Add files using upload-large-folder tool

README.md

@@ -1,5 +1,16 @@
 ---
 license: apache-2.0
+library_name: diffusers
+pipeline_tag: image-to-image
+tags:
+  - controlnet
+  - remote-sensing
+  - openstreetmap
+widget:
+  - src: demo_images/input.jpeg
+    prompt: convert this openstreetmap into its satellite view
+    output:
+      url: demo_images/output.jpeg
 ---
 # ControlEarth
 

inference_demo.py

@@ -19,6 +19,7 @@ def main():
     controlnet_path = repo / "controlnet"
     demo_images = repo / "demo_images"
     out_dir = repo / "outputs"
+    demo_out = repo / "demo_images"
     out_dir.mkdir(exist_ok=True)
 
     print("Loading ControlNet...")
@@ -70,6 +71,10 @@ def main():
             out_path = out_dir / f"{name}-{i}.png"
             result.save(out_path)
             print(f"  Saved {out_path}")
+            if i == 0:
+                demo_out.mkdir(exist_ok=True)
+                result.save(demo_out / "output.jpeg")
+                print(f"  Saved demo {demo_out / 'output.jpeg'}")
 
     print(f"Done. Outputs in {out_dir}")