Spaces:
Runtime error
Runtime error
Commit
·
a3bbfb3
1
Parent(s):
b264503
Upload 2 files
Browse files- app.py +1 -1
- 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-
|
| 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
|