ahamedddd commited on
Commit
30eb8c5
·
1 Parent(s): f16c38b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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"]