Sina1138 commited on
Commit
ab4e448
·
1 Parent(s): b9e007f

update polarity model path to use Sina1138/Scibert_polarity_Review

Browse files
Files changed (1) hide show
  1. glimpse-ui/interface/Demo.py +1 -1
glimpse-ui/interface/Demo.py CHANGED
@@ -123,7 +123,7 @@ def summarize(text1, text2, text3, focus, mode, rationality=1.0, iterations=1):
123
  sentences = list(set(text1_sentences + text2_sentences + text3_sentences))
124
 
125
  # Load polarity model and tokenizer (SciBERT)
126
- polarity_model_path = "scibert/scibert_polarity/final_model"
127
  polarity_tokenizer = AutoTokenizer.from_pretrained(polarity_model_path)
128
  polarity_model = AutoModelForSequenceClassification.from_pretrained(polarity_model_path)
129
  polarity_model.eval()
 
123
  sentences = list(set(text1_sentences + text2_sentences + text3_sentences))
124
 
125
  # Load polarity model and tokenizer (SciBERT)
126
+ polarity_model_path = "Sina1138/Scibert_polarity_Review"
127
  polarity_tokenizer = AutoTokenizer.from_pretrained(polarity_model_path)
128
  polarity_model = AutoModelForSequenceClassification.from_pretrained(polarity_model_path)
129
  polarity_model.eval()