RitaParadaRamos commited on
Commit
a3bbfb3
·
1 Parent(s): b264503

Upload 2 files

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
23
  feature_extractor = CLIPFeatureExtractor.from_pretrained("openai/clip-vit-base-patch32")
24
 
25
  # load and configure tokenizer
26
- tokenizer = AutoTokenizer.from_pretrained("facebook/opt-125M")
27
  tokenizer.pad_token = '!'
28
  tokenizer.eos_token = '.'
29
 
 
23
  feature_extractor = CLIPFeatureExtractor.from_pretrained("openai/clip-vit-base-patch32")
24
 
25
  # load and configure tokenizer
26
+ tokenizer = AutoTokenizer.from_pretrained("facebook/opt-125m")
27
  tokenizer.pad_token = '!'
28
  tokenizer.eos_token = '.'
29
 
requirements.txt CHANGED
@@ -12,7 +12,7 @@ torch==1.12.1
12
  torchaudio==0.12.1
13
  torchvision==0.13.1
14
  tqdm==4.64.0
15
- transformers
16
  ftfy
17
  regex
18
  tqdm
 
12
  torchaudio==0.12.1
13
  torchvision==0.13.1
14
  tqdm==4.64.0
15
+ transformers==4.21.1
16
  ftfy
17
  regex
18
  tqdm