Update app.py

app.py

@@ -1,3 +1,4 @@
+import os
 import torch
 from PIL import Image
 from torchvision import transforms