Update app.py
Browse files
app.py
CHANGED
|
@@ -126,11 +126,11 @@ def resize_image(image, resolution):
|
|
| 126 |
|
| 127 |
|
| 128 |
def preprocess_canny(image, resolution=128):
|
| 129 |
-
processed_image = cv2.Canny(
|
| 130 |
processed_image = processed_image[:, :, None]
|
| 131 |
processed_image = np.concatenate([processed_image, processed_image, processed_image], axis=2)
|
| 132 |
|
| 133 |
-
resized_image = Image.fromarray(
|
| 134 |
processed_image = Image.fromarray(processed_image)
|
| 135 |
return resized_image, processed_image
|
| 136 |
|
|
|
|
| 126 |
|
| 127 |
|
| 128 |
def preprocess_canny(image, resolution=128):
|
| 129 |
+
processed_image = cv2.Canny(image, low_threshold, high_threshold)
|
| 130 |
processed_image = processed_image[:, :, None]
|
| 131 |
processed_image = np.concatenate([processed_image, processed_image, processed_image], axis=2)
|
| 132 |
|
| 133 |
+
resized_image = Image.fromarray(image)
|
| 134 |
processed_image = Image.fromarray(processed_image)
|
| 135 |
return resized_image, processed_image
|
| 136 |
|