Spaces:

efeperro
/

Image_Captioning

Sleeping

efeperro commited on May 1, 2024

Commit

c8282e2

verified ·

1 Parent(s): 9c78b5c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,6 +5,7 @@ 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):

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