Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -31,7 +31,7 @@ controlnet = ControlNetModel.from_pretrained(
|
|
31 |
)
|
32 |
|
33 |
controlnet_lineart = ControlNetModel.from_pretrained(
|
34 |
-
"
|
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:
|