Update app.py

app.py

@@ -134,10 +134,10 @@ def infer(use_custom_model, model_name, weight_name, custom_lora_weight, image_i
         )
 
     if preprocessor == "lineart":
-        image = Image.open(image_in)
-        image = image.convert("RGB")
         image = np.array(image)
-        image = 255 - image
+        image = cv2.Canny(image, 100, 200)
+        image = image[:, :, None]
+        image = np.concatenate([image, image, image], axis=2)
         image = Image.fromarray(image)
 
         controlnet = ControlNetModel.from_pretrained(
@@ -162,8 +162,8 @@ def infer(use_custom_model, model_name, weight_name, custom_lora_weight, image_i
         controlnet=controlnet,
         vae=vae,
         torch_dtype=torch.float16, 
-        variant="fp16",
-        use_safetensors=True
+        #variant="fp16",
+        #use_safetensors=True
     )
     
     pipe.to(device)