Spaces:

efeperro
/

Image_Captioning

Sleeping

efeperro commited on May 1

Commit

59e90bd

•

1 Parent(s): 7cb104b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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

 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):