Update app.py

app.py

@@ -149,6 +149,19 @@ def infer(use_custom_model, model_name, weight_name, custom_lora_weight, image_i
         image = image[:, :, None]
         image = np.concatenate([image, image, image], axis=2)
         image = Image.fromarray(image)
+
+    if preprocessor == "lineart":
+        image = Image.open(image_in)
+
+        # Convert the image to a NumPy array
+        image = np.array(image)
+        
+        # (Optional) If you need to ensure the image is in RGB format, convert it
+        image = image.convert("RGB")
+        image = np.array(image)
+        
+        # Convert the NumPy array back to a PIL Image (if you need to do further PIL-based processing)
+        image = Image.fromarray(image)
     
     if use_custom_model: