multimodalart HF Staff commited on
Commit
4025aa9
·
verified ·
1 Parent(s): fc7c434

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -134,7 +134,7 @@ def get_control_image(input_image, control_mode):
134
  if control_mode == "Canny":
135
  return extract_canny(input_image)
136
  elif control_mode == "Soft Edge":
137
- return anyline(input_image, to_pil=True)
138
  elif control_mode == "Depth":
139
  image_np = np.array(input_image)
140
  with torch.no_grad():
 
134
  if control_mode == "Canny":
135
  return extract_canny(input_image)
136
  elif control_mode == "Soft Edge":
137
+ return anyline(input_image)
138
  elif control_mode == "Depth":
139
  image_np = np.array(input_image)
140
  with torch.no_grad():