Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -9,6 +9,8 @@ from timeit import default_timer as timer
|
|
9 |
from torchdata.datapipes.iter import IterableWrapper
|
10 |
from torch.utils.data import DataLoader
|
11 |
import torchtext.functional as F
|
|
|
|
|
12 |
|
13 |
# Setup class names
|
14 |
class_names = ["Bad", "Good"]
|
|
|
9 |
from torchdata.datapipes.iter import IterableWrapper
|
10 |
from torch.utils.data import DataLoader
|
11 |
import torchtext.functional as F
|
12 |
+
import torchtext.transforms as T
|
13 |
+
from torch.hub import load_state_dict_from_url
|
14 |
|
15 |
# Setup class names
|
16 |
class_names = ["Bad", "Good"]
|