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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -154,6 +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 = Image.fromarray(image)
158
 
159
  if use_custom_model:
 
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: