jugalgajjar commited on
Commit
3407f20
·
verified ·
1 Parent(s): 356c3ca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import torch
4
  import torch.nn as nn
5
  import torch.nn.functional as F
6
  import numpy as np
7
- from torchvision import transforms
8
 
9
  class_names = ['airplane', 'automobile', 'bird', 'cat', 'deer', 'dog', 'frog', 'horse', 'ship', 'truck']
10
 
 
4
  import torch.nn as nn
5
  import torch.nn.functional as F
6
  import numpy as np
7
+ import torchvision.transforms as T
8
 
9
  class_names = ['airplane', 'automobile', 'bird', 'cat', 'deer', 'dog', 'frog', 'horse', 'ship', 'truck']
10