Spaces:

efeperro
/

Image_Captioning

Sleeping

efeperro commited on May 1

Commit

aed55cf

•

1 Parent(s): efa8597

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,6 +4,7 @@ from PIL import Image
 from torchvision import transforms
 from transformers import T5Tokenizer, ViTFeatureExtractor
 from torch import nn
 class Encoder(nn.Module):
     def __init__(self, pretrained_model):

 from torchvision import transforms
 from transformers import T5Tokenizer, ViTFeatureExtractor
 from torch import nn
+from typing import Tuple
 class Encoder(nn.Module):
     def __init__(self, pretrained_model):