Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -26,7 +26,7 @@ def print_bn():
|
|
26 |
|
27 |
def update_bn(image):
|
28 |
cursor_im = 0
|
29 |
-
image = T.Resize((
|
30 |
image = image.reshape(-1)
|
31 |
for m in model.modules():
|
32 |
if(type(m) is nn.BatchNorm2d):
|
|
|
26 |
|
27 |
def update_bn(image):
|
28 |
cursor_im = 0
|
29 |
+
image = T.Resize((90,90))(image)
|
30 |
image = image.reshape(-1)
|
31 |
for m in model.modules():
|
32 |
if(type(m) is nn.BatchNorm2d):
|