ovshake commited on
Commit
55f0240
1 Parent(s): ae6d3bd

change app

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -94,6 +94,9 @@ def process_image(args, inpainting_pipeline, net):
94
  output_arr = output_tensor.cpu().numpy()
95
  mask_code = eval(f"Parts.{args.part.upper()}")
96
  mask = (output_arr == mask_code)
 
 
 
97
  output_arr[mask] = 1
98
  output_arr[~mask] = 0
99
  output_arr *= 255
 
94
  output_arr = output_tensor.cpu().numpy()
95
  mask_code = eval(f"Parts.{args.part.upper()}")
96
  mask = (output_arr == mask_code)
97
+ print(f"Numbers in output_rr")
98
+ print(np.unique(output_arr))
99
+ print(f"mask code {mask_code}")
100
  output_arr[mask] = 1
101
  output_arr[~mask] = 0
102
  output_arr *= 255