Update app.py

app.py

@@ -5,7 +5,10 @@ import torch
 from model import TinyCNN
 from timeit import default_timer as timer
 from typing import Tuple, Dict
+import torch
+import torchvision
 
+from torch import nn
 # Setup class names
 with open("class_names.txt", "r") as f: # reading them in from class_names.txt
     class_names = [food_name.strip() for food_name in  f.readlines()]