Update app.py

app.py

@@ -10,6 +10,7 @@ from pathlib import Path
 import random
 
 import PIL
+import torchvision.transforms as transforms
 
 import gradio as gr