Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -134,6 +134,7 @@ def infer(use_custom_model, model_name, weight_name, custom_lora_weight, image_i
|
|
| 134 |
)
|
| 135 |
|
| 136 |
if preprocessor == "lineart":
|
|
|
|
| 137 |
image = np.array(image)
|
| 138 |
image = cv2.Canny(image, 100, 200)
|
| 139 |
image = image[:, :, None]
|
|
|
|
| 134 |
)
|
| 135 |
|
| 136 |
if preprocessor == "lineart":
|
| 137 |
+
image = load_image(image_in)
|
| 138 |
image = np.array(image)
|
| 139 |
image = cv2.Canny(image, 100, 200)
|
| 140 |
image = image[:, :, None]
|