RitaParadaRamos commited on
Commit
fe53098
·
1 Parent(s): a1eae7e

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +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