dnnsdunca commited on
Commit
d58ba2c
1 Parent(s): be8f65b

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +11 -1
requirements.txt CHANGED
@@ -1,3 +1,13 @@
1
  transformers==4.20.1
2
  torch==1.12.0
3
- pandas==1.4.2
 
 
 
 
 
 
 
 
 
 
 
1
  transformers==4.20.1
2
  torch==1.12.0
3
+ pandas==1.4.2
4
+ git clone https://huggingface.co/google-bert/bert-base-uncased
5
+ # Load model directly
6
+ from transformers import AutoTokenizer, AutoModelForMaskedLM
7
+
8
+ tokenizer = AutoTokenizer.from_pretrained("google-bert/bert-base-uncased")
9
+ model = AutoModelForMaskedLM.from_pretrained("google-bert/bert-base-uncased")
10
+ # Use a pipeline as a high-level helper
11
+ from transformers import pipeline
12
+
13
+ pipe = pipeline("fill-mask", model="google-bert/bert-base-uncased")