Changed repo name

README.md

@@ -61,7 +61,7 @@ class VITContrastiveHF(nn.Module):
     
 # HF inference code
 classificator_type = 'linear'
-model = VITContrastiveHF(repo_name='fede97/CoDE', classificator_type=classificator_type)
+model = VITContrastiveHF(repo_name='aimagelab/CoDE', classificator_type=classificator_type)
 
 transform = transforms.Compose([
     transforms.CenterCrop(224),