theSure commited on
Commit
ba3fee7
·
verified ·
1 Parent(s): 56c7591

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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] % 8)
115
- H = int(np.shape(img)[1] - np.shape(img)[1] % 8)
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))