rrevoid commited on
Commit
89d4e60
·
1 Parent(s): 30fb1b6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,10 +1,11 @@
1
  import torch
 
2
 
3
  import streamlit as st
4
  import torch.nn as nn
5
  from transformers import RobertaTokenizer, RobertaModel
6
 
7
- @st.cache(suppress_st_warning=True)
8
  def init():
9
  tokenizer = RobertaTokenizer.from_pretrained("roberta-large-mnli")
10
 
 
1
  import torch
2
+ import tokenizers
3
 
4
  import streamlit as st
5
  import torch.nn as nn
6
  from transformers import RobertaTokenizer, RobertaModel
7
 
8
+ @st.cache(hash_funcs={tokenizers.Tokenizer: lambda _: None})
9
  def init():
10
  tokenizer = RobertaTokenizer.from_pretrained("roberta-large-mnli")
11