Spaces:

naver
/

SuperFeatures

Build error

YannisK commited on Apr 21, 2022

Commit

0f68393

•

1 Parent(s): 1d018e4

edits

Files changed (1) hide show

app.py CHANGED Viewed

@@ -45,7 +45,7 @@ net_imagenet.load_state_dict(state2['state_dict'], strict=False)
 transform = transforms.Compose([
         transforms.Resize(1024),
         transforms.ToTensor(),
-        transforms.Normalize(**dict(zip(["mean", "std"], net.runtime['mean_std'])))
         ])
 # ---------------------------------------

 transform = transforms.Compose([
         transforms.Resize(1024),
         transforms.ToTensor(),
+        transforms.Normalize(**dict(zip(["mean", "std"], net_sfm.runtime['mean_std'])))
         ])
 # ---------------------------------------