hitz02 commited on
Commit
bc819a0
·
1 Parent(s): a594bcf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -34,7 +34,7 @@ def load_prep_data():
34
  @st.cache(allow_output_mutation=True)
35
  def build_sent_trans_model():
36
 
37
- word_embedding_model = models.BERT('./covidbert_nli')
38
 
39
  # Add the pooling strategy of Mean
40
  pooling_model = models.Pooling(word_embedding_model.get_word_embedding_dimension(),
 
34
  @st.cache(allow_output_mutation=True)
35
  def build_sent_trans_model():
36
 
37
+ word_embedding_model = models.BERT('./')
38
 
39
  # Add the pooling strategy of Mean
40
  pooling_model = models.Pooling(word_embedding_model.get_word_embedding_dimension(),