fffiloni commited on
Commit
4875fec
1 Parent(s): c87e32a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -148,7 +148,8 @@ def infer(use_custom_model, model_name, weight_name, custom_lora_weight, image_i
148
  image = np.concatenate([image, image, image], axis=2)
149
  image = Image.fromarray(image)
150
  elif preprocessor == "lineart":
151
- image = PIL.Image.fromarray(image_in)
 
152
 
153
  if use_custom_model:
154
 
 
148
  image = np.concatenate([image, image, image], axis=2)
149
  image = Image.fromarray(image)
150
  elif preprocessor == "lineart":
151
+ image = load_image(image_in)
152
+ image = Image.fromarray(image)
153
 
154
  if use_custom_model:
155