Update app.py

app.py

@@ -17,7 +17,6 @@ from diffusers import EulerDiscreteScheduler
 from PIL import Image, ImageDraw, ImageFont
 import os
 
-
 device = "cuda"
 ckpt_dir = snapshot_download(repo_id="Kwai-Kolors/Kolors")
 ckpt_dir_canny = snapshot_download(repo_id="Kwai-Kolors/Kolors-ControlNet-Canny")
@@ -49,9 +48,27 @@ def translate_korean_to_english(text):
         return translated
     return text
 
+def HWC3(x):
+    assert x.dtype == np.uint8
+    if x.ndim == 2:
+        x = x[:, :, None]
+    assert x.ndim == 3
+    H, W, C = x.shape
+    assert C == 1 or C == 3 or C == 4
+    if C == 3:
+        return x
+    if C == 1:
+        return np.concatenate([x, x, x], axis=2)
+    if C == 4:
+        color = x[:, :, 0:3].astype(np.float32)
+        alpha = x[:, :, 3:4].astype(np.float32) / 255.0
+        y = color * alpha + 255.0 * (1.0 - alpha)
+        y = y.clip(0, 255).astype(np.uint8)
+        return y
+
 @spaces.GPU
 def process_canny_condition(image, canny_threods=[100,200]):
-    np_image = image.copy()
+    np_image = np.array(image)
     np_image = cv2.Canny(np_image, canny_threods[0], canny_threods[1])
     np_image = np_image[:, :, None]
     np_image = np.concatenate([np_image, np_image, np_image], axis=2)
@@ -61,6 +78,13 @@ def process_canny_condition(image, canny_threods=[100,200]):
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024
 
+def resize_image(image, resolution):
+    w, h = image.size
+    ratio = resolution / max(w, h)
+    new_w = int(w * ratio)
+    new_h = int(h * ratio)
+    return image.resize((new_w, new_h), Image.LANCZOS)
+
 def text_to_image(text, size=72, position="middle-center"):
     width, height = 1024, 576
     image = Image.new("RGB", (width, height), "white")
@@ -112,7 +136,6 @@ def text_to_image(text, size=72, position="middle-center"):
 
     return image
 
-
 @spaces.GPU
 def infer_canny(prompt, text_for_image, 
           negative_prompt = "nsfw, facial shadows, low resolution, jpeg artifacts, blurry, bad quality, dark face, neon lights", 
@@ -136,7 +159,7 @@ def infer_canny(prompt, text_for_image,
     init_image = resize_image(init_image, MAX_IMAGE_SIZE)
     
     pipe = pipe_canny.to("cuda")
-    condi_img = process_canny_condition(np.array(init_image))
+    condi_img = process_canny_condition(init_image)
     image = pipe(
         prompt=prompt,
         image=init_image,
@@ -239,4 +262,4 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as Kolors:
         outputs = [result, seed_used]
     )
 
-Kolors.queue().launch(debug=True)
+Kolors.queue().launch(debug=True, share=True)