fffiloni commited on
Commit
1451797
1 Parent(s): 1445d23

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -9
app.py CHANGED
@@ -31,8 +31,9 @@ controlnet = ControlNetModel.from_pretrained(
31
  )
32
 
33
  controlnet_lineart = ControlNetModel.from_pretrained(
34
- "r3gm/controlnet-lineart-anime-sdxl-fp16",
35
- torch_dtype=torch.float16
 
36
  )
37
 
38
  def check_use_custom_or_no(value):
@@ -149,13 +150,6 @@ def infer(use_custom_model, model_name, weight_name, custom_lora_weight, image_i
149
  image = np.concatenate([image, image, image], axis=2)
150
  image = Image.fromarray(image)
151
 
152
- if preprocessor == "lineart":
153
- image = load_image(image_in)
154
- image = np.array(image)
155
- image = HWC3(image)
156
- #image = resize_image(image, resolution=image_resolution)
157
- image = Image.fromarray(image)
158
-
159
  if use_custom_model:
160
 
161
  if model_name == "":
 
31
  )
32
 
33
  controlnet_lineart = ControlNetModel.from_pretrained(
34
+ "TheMistoAI/MistoLine",
35
+ torch_dtype=torch.float16,
36
+ variant="fp16",
37
  )
38
 
39
  def check_use_custom_or_no(value):
 
150
  image = np.concatenate([image, image, image], axis=2)
151
  image = Image.fromarray(image)
152
 
 
 
 
 
 
 
 
153
  if use_custom_model:
154
 
155
  if model_name == "":