Update app.py

app.py

@@ -2,6 +2,7 @@
 import gradio as gr
 import torch
 from torchvision import models, transforms
+from PIL import Image
 '''
 # -- get torch and cuda version
 TORCH_VERSION = ".".join(torch.__version__.split(".")[:2])