Spaces:

efeperro
/

Image_Captioning

Sleeping

efeperro commited on May 1, 2024

Commit

1ec1322

verified ·

1 Parent(s): 9633fb6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,7 +2,7 @@ import gradio as gr
 import torch
 from PIL import Image
 from torchvision import transforms
-from transformers import T5Tokenizer, ViTFeatureExtractor
 from torch import nn
 from typing import Tuple
 import sentencepiece

 import torch
 from PIL import Image
 from torchvision import transforms
+from transformers import T5Tokenizer, ViTImageProcessor
 from torch import nn
 from typing import Tuple
 import sentencepiece