Baptlem commited on
Commit
783c45d
1 Parent(s): 5461399

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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(resized_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(resized_image)
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