Update app.py

app.py

@@ -1,6 +1,6 @@
 import torch
 import torch.nn as nn
-from torchvision import transforms
+from torchvision import transforms, models
 from PIL import Image
 import requests
 from torchvision.models import vgg19