Spaces:
Runtime error
Runtime error
RamAnanth1
commited on
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)
|