Update app.py

app.py

@@ -7,7 +7,7 @@ from timeit import default_timer as timer
 from typing import Tuple, Dict
 import torch
 import torchvision
-
+from torchvision import transforms
 from torch import nn
 # Setup class names
 with open("class_names.txt", "r") as f: # reading them in from class_names.txt