Spaces:
Running
Running
Fix
Browse files
model.py
CHANGED
@@ -542,6 +542,10 @@ class Model:
|
|
542 |
control_image, _ = apply_midas(
|
543 |
resize_image(input_image, detect_resolution))
|
544 |
control_image = HWC3(control_image)
|
|
|
|
|
|
|
|
|
545 |
return PIL.Image.fromarray(control_image), PIL.Image.fromarray(
|
546 |
control_image)
|
547 |
|
|
|
542 |
control_image, _ = apply_midas(
|
543 |
resize_image(input_image, detect_resolution))
|
544 |
control_image = HWC3(control_image)
|
545 |
+
image = resize_image(input_image, image_resolution)
|
546 |
+
H, W = image.shape[:2]
|
547 |
+
control_image = cv2.resize(control_image, (W, H),
|
548 |
+
interpolation=cv2.INTER_LINEAR)
|
549 |
return PIL.Image.fromarray(control_image), PIL.Image.fromarray(
|
550 |
control_image)
|
551 |
|