Spaces:
Runtime error
Runtime error
ovshake
commited on
Commit
•
55f0240
1
Parent(s):
ae6d3bd
change app
Browse files
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
|