Spaces:

Wootang01
/

image_classifier

Runtime error

Wootang01 commited on Mar 16, 2022

Commit

3d7e8cb

1 Parent(s): 1087fcb

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ import torch
 from timm import create_model
 from timm.data import resolve_data_config
-from timm.data.transform import create_transform
 IMAGENET_1k_URL = "https://storage.googleapis.com/bit_models/ilsvrc2012_wordnet_lemmas.txt"
 LABELS = requests.get(IMAGENET_1k_URL).text.strip().split('\n')

 from timm import create_model
 from timm.data import resolve_data_config
+from timm.data.transforms_factory import create_transform
 IMAGENET_1k_URL = "https://storage.googleapis.com/bit_models/ilsvrc2012_wordnet_lemmas.txt"
 LABELS = requests.get(IMAGENET_1k_URL).text.strip().split('\n')