SajjadAyoubi commited on
Commit
9acbb96
·
1 Parent(s): 90ecddb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -7,9 +7,7 @@ import torch
7
  import transformers
8
  from transformers import RobertaModel, AutoTokenizer
9
 
10
- # @st.cache(hash_funcs={transformers.models.gpt2.tokenization_gpt2_fast.GPT2TokenizerFast: hash}, suppress_st_warning=True, allow_output_mutation=True)
11
- @st.cache(show_spinner=False, hash_funcs={transformers.models.roberta.tokenization_roberta_fast.RobertaTokenizerFast: hash},
12
- suppress_st_warning=True, allow_output_mutation=True)
13
  def load():
14
  text_encoder = RobertaModel.from_pretrained('SajjadAyoubi/clip-fa-text')
15
  tokenizer = AutoTokenizer.from_pretrained('SajjadAyoubi/clip-fa-text')
 
7
  import transformers
8
  from transformers import RobertaModel, AutoTokenizer
9
 
10
+ # @st.cache(hash_funcs={transformers.models.roberta.tokenization_roberta_fast.RobertaTokenizerFast: hash}, suppress_st_warning=True, allow_output_mutation=True)
 
 
11
  def load():
12
  text_encoder = RobertaModel.from_pretrained('SajjadAyoubi/clip-fa-text')
13
  tokenizer = AutoTokenizer.from_pretrained('SajjadAyoubi/clip-fa-text')