Spaces:

Grosy
/

RF_QA

Runtime error

Grosy commited on May 12, 2022

Commit

8eaf3f5

•

1 Parent(s): 5bec655

Fixed link

Files changed (1) hide show

app.py CHANGED Viewed

@@ -55,7 +55,7 @@ def load_model_and_tokenizer():
 @st.cache(hash_funcs={transformers.models.bert.tokenization_bert_fast.BertTokenizerFast: lambda _: None, transformers.models.bert.modeling_bert.BertModel: lambda _: None})
 def load_hu_model_and_tokenizer():
-    multilingual_checkpoint = 'sentence-transformers/SZTAKI-HLT/hubert-base-cc'  #alternative: SZTAKI-HLT/hubert-base-cc
     tokenizer = AutoTokenizer.from_pretrained(multilingual_checkpoint)
     model = AutoModel.from_pretrained(multilingual_checkpoint)
     print(type(tokenizer))

 @st.cache(hash_funcs={transformers.models.bert.tokenization_bert_fast.BertTokenizerFast: lambda _: None, transformers.models.bert.modeling_bert.BertModel: lambda _: None})
 def load_hu_model_and_tokenizer():
+    multilingual_checkpoint = 'SZTAKI-HLT/hubert-base-cc'  #alternative: SZTAKI-HLT/hubert-base-cc
     tokenizer = AutoTokenizer.from_pretrained(multilingual_checkpoint)
     model = AutoModel.from_pretrained(multilingual_checkpoint)
     print(type(tokenizer))