baimamboukar commited on
Commit
198eb21
1 Parent(s): daa014f

chore(fix): fix model name

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ from timm.data.transforms_factory import create_transform
10
  IMAGENET_1K_URL = 'https://storage.googleapis.com/bit_models/ilsvrc2012_1k_wordnet_lemmas.txt'
11
  IMAGENET_1K_LABELS = requests.get(IMAGENET_1K_URL).text.strip().split('\n')
12
 
13
- model = create_model('restnet50', pretrained=True)
14
 
15
  transform = create_transform(
16
  **resolve_data_config({}, model=model)['test_time_augmentation'][0])
 
10
  IMAGENET_1K_URL = 'https://storage.googleapis.com/bit_models/ilsvrc2012_1k_wordnet_lemmas.txt'
11
  IMAGENET_1K_LABELS = requests.get(IMAGENET_1K_URL).text.strip().split('\n')
12
 
13
+ model = create_model('resnet50', pretrained=True)
14
 
15
  transform = create_transform(
16
  **resolve_data_config({}, model=model)['test_time_augmentation'][0])