Update app.py

app.py

@@ -1,6 +1,7 @@
 import cv2, torch 
 import gradio as gr
 import numpy as np 
+from PIL import Image 
 import torch.nn as nn
 import torchvision.models as models 
 from torchvision import transforms as T