SarmadBashir commited on
Commit
8da0593
β€’
1 Parent(s): ef8be1a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ st.set_page_config(
8
  #page_icon="🎈",
9
  )
10
 
11
- @st.cache
12
  def get_pipleine():
13
  tokenizer = AutoTokenizer.from_pretrained("SarmadBashir/dronology_bert_uncased", model_max_length = 128)
14
  model = AutoModelForSequenceClassification.from_pretrained("SarmadBashir/dronology_bert_uncased")
 
8
  #page_icon="🎈",
9
  )
10
 
11
+ @st.cache(allow_output_mutation=True)
12
  def get_pipleine():
13
  tokenizer = AutoTokenizer.from_pretrained("SarmadBashir/dronology_bert_uncased", model_max_length = 128)
14
  model = AutoModelForSequenceClassification.from_pretrained("SarmadBashir/dronology_bert_uncased")