Create app.py

app.py

@@ -0,0 +1,30 @@
+import scripts.control_utils as cu
+import torch
+from PIL import Image
+
+path_to_config = 'ControlNet-XS-main/configs/inference/sdxl/sdxl_encD_canny_48m.yaml'
+model = cu.create_model(path_to_config).to('cuda')
+
+image_path = 'PATH/TO/IMAGES/Shoe.png'
+
+canny_high_th = 250
+canny_low_th = 100
+size = 768
+num_samples=2
+
+image = cu.get_image(image_path, size=size)
+edges = cu.get_canny_edges(image, low_th=canny_low_th, high_th=canny_high_th)
+
+samples, controls = cu.get_sdxl_sample(
+    guidance=edges,
+    ddim_steps=10,
+    num_samples=num_samples,
+    model=model,
+    shape=[4, size // 8, size // 8],
+    control_scale=0.95,
+    prompt='cinematic, shoe in the streets, made from meat, photorealistic shoe, highly detailed',
+    n_prompt='lowres, bad anatomy, worst quality, low quality',
+)
+
+
+Image.fromarray(cu.create_image_grid(samples)).save('SDXL_MyShoe.png')