ngaggion commited on
Commit
0486f3c
·
1 Parent(s): f428da1
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -146,6 +146,8 @@ def removePreprocess(output, info):
146
 
147
  if h != 1024 or w != 1024:
148
  output = output * h
 
 
149
 
150
  padh, padw, auxh, auxw = padding
151
 
 
146
 
147
  if h != 1024 or w != 1024:
148
  output = output * h
149
+ else:
150
+ output = output * 1024
151
 
152
  padh, padw, auxh, auxw = padding
153