Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
141 |
image = Image.fromarray(image)
|
142 |
|
143 |
controlnet = ControlNetModel.from_pretrained(
|
144 |
-
"
|
145 |
torch_dtype=torch.float16,
|
146 |
-
|
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(
|