Spaces:
Running
on
A10G
Running
on
A10G
Update app.py
Browse files
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 =
|
|
|
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 |
|