paragon-analytics commited on
Commit
545fde2
1 Parent(s): a75e0ed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -107,7 +107,7 @@ def process_final_text(text):
107
  # generated_ids = para_model.generate(batch['input_ids'])
108
  # para_list = para_tokenizer.batch_decode(generated_ids, skip_special_tokens=True)
109
 
110
- return {"Resilience": float(scores.numpy()[1]), "Non-Resilience": float(scores.numpy()[0])},keywords,NER,word_attributions
111
 
112
  def main(prob1):
113
  text = str(prob1)
 
107
  # generated_ids = para_model.generate(batch['input_ids'])
108
  # para_list = para_tokenizer.batch_decode(generated_ids, skip_special_tokens=True)
109
 
110
+ return {"Inspirational": float(scores.numpy()[1]), "Uninspiring": float(scores.numpy()[0])},keywords,NER,word_attributions
111
 
112
  def main(prob1):
113
  text = str(prob1)