fffiloni commited on
Commit
643afc0
1 Parent(s): 1f92b86

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -137,13 +137,13 @@ def infer(use_custom_model, model_name, weight_name, custom_lora_weight, image_i
137
  image = Image.open(image_in)
138
  image = image.convert("RGB")
139
  image = np.array(image)
140
- #image = 255 - image
141
  image = Image.fromarray(image)
142
 
143
  controlnet = ControlNetModel.from_pretrained(
144
- "fffiloni/cn_malgras_test_11",
145
  torch_dtype=torch.float16,
146
- token=hf_token
147
  )
148
 
149
  pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
 
137
  image = Image.open(image_in)
138
  image = image.convert("RGB")
139
  image = np.array(image)
140
+ image = 255 - image
141
  image = Image.fromarray(image)
142
 
143
  controlnet = ControlNetModel.from_pretrained(
144
+ "TheMistoAI/MistoLine",
145
  torch_dtype=torch.float16,
146
+ variant="fp16"
147
  )
148
 
149
  pipe = StableDiffusionXLControlNetPipeline.from_pretrained(