fffiloni commited on
Commit
2efa876
1 Parent(s): 715f9ff

Update app.py

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