Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -108,7 +108,7 @@ def get_control_image(input_image, control_mode):
|
|
108 |
if control_mode == "Canny":
|
109 |
return extract_canny(input_image)
|
110 |
elif control_mode == "Soft Edge":
|
111 |
-
return
|
112 |
elif control_mode == "Depth":
|
113 |
image_np = np.array(input_image)
|
114 |
with torch.no_grad():
|
|
|
108 |
if control_mode == "Canny":
|
109 |
return extract_canny(input_image)
|
110 |
elif control_mode == "Soft Edge":
|
111 |
+
return anyline(input_image, to_pil=True)
|
112 |
elif control_mode == "Depth":
|
113 |
image_np = np.array(input_image)
|
114 |
with torch.no_grad():
|