fffiloni commited on
Commit
2471e0c
·
verified ·
1 Parent(s): 24ff7da

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -5
app.py CHANGED
@@ -149,11 +149,7 @@ def infer(use_custom_model, model_name, weight_name, custom_lora_weight, image_i
149
  image = Image.fromarray(image)
150
 
151
  if preprocessor == "lineart":
152
- image = load_image(image_in)
153
- image = np.array(image)
154
- image = image[:, :, None]
155
- image = np.concatenate([image, image, image], axis=2)
156
- image = Image.fromarray(image)
157
 
158
  if use_custom_model:
159
 
 
149
  image = Image.fromarray(image)
150
 
151
  if preprocessor == "lineart":
152
+ image = Image.open(image_in)
 
 
 
 
153
 
154
  if use_custom_model:
155