resolved icon issue

README.md

@@ -1,6 +1,6 @@
 ---
 title: Watermarking in SSL latent spaces
-emoji: :lock:
+emoji: 🔓
 colorFrom: indigo
 colorTo: red
 sdk: gradio

app.py

@@ -14,12 +14,12 @@ torch.manual_seed(0)
 np.random.seed(0)
 
 FPR = 1e-6
-carrier = np.random.randn(size=(1, 2048))
+carrier = np.random.randn(1, 2048)
 
 
 def build_backbone(path, name='resnet50'):
     """ Builds a pretrained ResNet-50 backbone. """
-    model = getattr(models, name)(pretrained=True)
+    model = getattr(models, name)(pretrained=False)
     model.head = nn.Identity()
     model.fc = nn.Identity()
     checkpoint = torch.load(path, map_location=device)