fix: fix model error

app.py

@@ -13,7 +13,7 @@ IMAGENET_1K_LABELS = requests.get(IMAGENET_1K_URL).text.strip().split('\n')
 model = create_model('resnet50', pretrained=True)
 
 transform = create_transform(
-    **resolve_data_config({}, model=model)['test_time_augmentation'][0])
+    **resolve_data_config({}, model=model))
 
 
 model.eval()