Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -111,8 +111,8 @@ def predict(
|
|
111 |
|
112 |
img = np.array(input_image.convert("RGB"))
|
113 |
|
114 |
-
W = int(np.shape(img)[0] - np.shape(img)[0] %
|
115 |
-
H = int(np.shape(img)[1] - np.shape(img)[1] %
|
116 |
|
117 |
input_image = input_image.resize((H, W))
|
118 |
uploaded_mask = uploaded_mask.resize((H, W))
|
|
|
111 |
|
112 |
img = np.array(input_image.convert("RGB"))
|
113 |
|
114 |
+
W = int(np.shape(img)[0] - np.shape(img)[0] % 16)
|
115 |
+
H = int(np.shape(img)[1] - np.shape(img)[1] % 16)
|
116 |
|
117 |
input_image = input_image.resize((H, W))
|
118 |
uploaded_mask = uploaded_mask.resize((H, W))
|