Spaces:
Runtime error
Runtime error
Commit
·
c71f561
1
Parent(s):
48a96cb
Update app.py
Browse files
app.py
CHANGED
|
@@ -48,7 +48,6 @@ pipe_pose.enable_xformers_memory_efficient_attention()
|
|
| 48 |
|
| 49 |
|
| 50 |
def get_canny_filter(image):
|
| 51 |
-
print(image)
|
| 52 |
|
| 53 |
if not isinstance(image, np.ndarray):
|
| 54 |
image = np.array(image)
|
|
@@ -67,8 +66,8 @@ def process(input_image, prompt, input_control):
|
|
| 67 |
# TODO: Add other control tasks
|
| 68 |
if input_control == "Pose":
|
| 69 |
return process_pose(input_image, prompt)
|
| 70 |
-
|
| 71 |
-
|
| 72 |
|
| 73 |
def process_canny(input_image, prompt):
|
| 74 |
canny_image = get_canny_filter(input_image)
|
|
|
|
| 48 |
|
| 49 |
|
| 50 |
def get_canny_filter(image):
|
|
|
|
| 51 |
|
| 52 |
if not isinstance(image, np.ndarray):
|
| 53 |
image = np.array(image)
|
|
|
|
| 66 |
# TODO: Add other control tasks
|
| 67 |
if input_control == "Pose":
|
| 68 |
return process_pose(input_image, prompt)
|
| 69 |
+
else:
|
| 70 |
+
return process_canny(input_image, prompt)
|
| 71 |
|
| 72 |
def process_canny(input_image, prompt):
|
| 73 |
canny_image = get_canny_filter(input_image)
|