Spaces:

bousejin
/

gradio-demo

Sleeping

bousejin commited on Mar 23, 2022

Commit

b1eeeff

•

1 Parent(s): 8963d31

fix bug

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ import gradio as gr
 import torch
 from timm import create_model
 from timm.data import resolve_data_config
-from timm.data.transforms 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')

 import torch
 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')