Spaces:

morinop
/

hellominori

Sleeping

morinop commited on Mar 3, 2023

Commit

079f211

•

1 Parent(s): 11dac02

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -24,8 +24,8 @@ def print_bn():
 def update_bn(image):
     cursor_im = 0
-    image = image.reshape(-1)
     image = T.Resize((40,40))(image)
     for m in model.modules():
         if(type(m) is nn.BatchNorm2d):
             if(cursor_im < image.shape[0]):

 def update_bn(image):
     cursor_im = 0
     image = T.Resize((40,40))(image)
+    image = image.reshape(-1)
     for m in model.modules():
         if(type(m) is nn.BatchNorm2d):
             if(cursor_im < image.shape[0]):