Update app.py

app.py

@@ -20,6 +20,8 @@ import clip
 from tqdm.notebook import tqdm
 from torchvision.transforms import Compose, Resize, ToTensor, Normalize
 from einops import rearrange
+import gradio as gr
+
 device = torch.device('cuda:0')
 def fetch(url_or_path):
     if str(url_or_path).startswith('http://') or str(url_or_path).startswith('https://'):