Hugging Face's logo

Spaces:
fuhsiao
/
Ext-Abs-StructuredSum
Runtime error

App Files Community
fuhsiao418 commited on
Commit
9072490
·
1 Parent(s): f6b2292

update

Browse files
Files changed (1) hide show
  1. utils/preprocess.py +1 -1
utils/preprocess.py CHANGED
@@ -143,7 +143,7 @@ def similarity(lst, ptm):
143
  def feature_extraction(title, section, sents):
144
  lst = sent_lst(sents)
145
  tfisf = Tfisf(lst)
146
- cosine = similarity(lst, "pritamdeka/PubMedBERT-mnli-snli-scinli-scitail-mednli-stsb")
147
 
148
  # Number of sentences
149
  ns = len(sents)
 
143
  def feature_extraction(title, section, sents):
144
  lst = sent_lst(sents)
145
  tfisf = Tfisf(lst)
146
+ cosine = similarity(lst, "sentence-transformers/all-MiniLM-L6-v2")
147
 
148
  # Number of sentences
149
  ns = len(sents)