Update app.py

app.py

@@ -2,6 +2,7 @@ import os
 from PIL import Image
 import torch
 from torchvision import transforms
+import gradio as gr
 
 
 # load model