fffiloni commited on
Commit
ee6e79a
·
verified ·
1 Parent(s): 36a5d43

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -31,7 +31,7 @@ controlnet = ControlNetModel.from_pretrained(
31
  )
32
 
33
  controlnet_lineart = ControlNetModel.from_pretrained(
34
- "TheMistoAI/MistoLine",
35
  torch_dtype=torch.float16,
36
  variant="fp16",
37
  )
@@ -154,7 +154,7 @@ def infer(use_custom_model, model_name, weight_name, custom_lora_weight, image_i
154
  image = Image.open(image_in)
155
  image = image.convert("RGB")
156
  image = np.array(image)
157
- image = 255 - image
158
  image = Image.fromarray(image)
159
 
160
  if use_custom_model:
 
31
  )
32
 
33
  controlnet_lineart = ControlNetModel.from_pretrained(
34
+ "fffiloni/cn_malgras_test_11",
35
  torch_dtype=torch.float16,
36
  variant="fp16",
37
  )
 
154
  image = Image.open(image_in)
155
  image = image.convert("RGB")
156
  image = np.array(image)
157
+ #image = 255 - image
158
  image = Image.fromarray(image)
159
 
160
  if use_custom_model: