Spaces:
Build error
Build error
fix bug
Browse files
app.py
CHANGED
|
@@ -4,7 +4,7 @@ import gradio as gr
|
|
| 4 |
import torch
|
| 5 |
from timm import create_model
|
| 6 |
from timm.data import resolve_data_config
|
| 7 |
-
from timm.data.
|
| 8 |
|
| 9 |
IMAGENET_1K_URL = 'https://storage.googleapis.com/bit_models/ilsvrc2012_wordnet_lemmas.txt'
|
| 10 |
LABELS = requests.get(IMAGENET_1K_URL).text.strip().split('\n')
|
|
|
|
| 4 |
import torch
|
| 5 |
from timm import create_model
|
| 6 |
from timm.data import resolve_data_config
|
| 7 |
+
from timm.data.transforms_factory import create_transform
|
| 8 |
|
| 9 |
IMAGENET_1K_URL = 'https://storage.googleapis.com/bit_models/ilsvrc2012_wordnet_lemmas.txt'
|
| 10 |
LABELS = requests.get(IMAGENET_1K_URL).text.strip().split('\n')
|