Update app.py

app.py

@@ -19,7 +19,7 @@ from torch import autocast
 from torchvision import transforms as tfms
 from tqdm.auto import tqdm
 from transformers import CLIPTextModel, CLIPTokenizer, logging
-
+import gradio as gr
 torch.manual_seed(1)
 #if not (Path.home()/'.huggingface'/'token').exists(): notebook_login()