Update app.py

app.py

@@ -1,3 +1,5 @@
+pip install torchvision
+
 import torch
 from torchvision import transforms
 import numpy as np