Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -150,6 +150,7 @@ def infer(use_custom_model, model_name, weight_name, custom_lora_weight, image_i
|
|
| 150 |
|
| 151 |
if preprocessor == "lineart":
|
| 152 |
image = load_image(image_in)
|
|
|
|
| 153 |
image = Image.fromarray(image)
|
| 154 |
|
| 155 |
if use_custom_model:
|
|
|
|
| 150 |
|
| 151 |
if preprocessor == "lineart":
|
| 152 |
image = load_image(image_in)
|
| 153 |
+
image = np.array(image)
|
| 154 |
image = Image.fromarray(image)
|
| 155 |
|
| 156 |
if use_custom_model:
|