Spaces:

Shrey
/

cc-bert

Sleeping

Shrey commited on Oct 19, 2022

Commit

136cb9d

1 Parent(s): 20223ca

added missing import

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,5 +1,5 @@
 #the inference function
-from transformers import FillMaskPipeline ,DistilBertTokenizer,TFAutoModelForMaskedLM
 from transformers import BertTokenizer
 #load the tokenizer

 #the inference function
+from transformers import FillMaskPipeline ,DistilBertTokenizer,TFAutoModelForMaskedLM,AutoTokenizer
 from transformers import BertTokenizer
 #load the tokenizer