RamAnanth1 commited on
Commit
c71f561
1 Parent(s): 48a96cb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
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
- return process_canny(input_image, prompt)
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)