Spaces:

Mayanand
/

Image-Captioning

Build error

Mayanand commited on Nov 26, 2022

Commit

716c14d

1 Parent(s): abd9184

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,6 +4,7 @@ import gradio as gr
 from io import BytesIO
 from PIL import Image
 from model import CaptionModel
 from preprocess import Tokenizer, return_user_agent
 tokenizer = Tokenizer('./')

 from io import BytesIO
 from PIL import Image
 from model import CaptionModel
+from torchvision import transforms
 from preprocess import Tokenizer, return_user_agent
 tokenizer = Tokenizer('./')