lalital commited on
Commit
f26f205
β€’
1 Parent(s): cf6195e

fix: set return_all_scores to true

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -41,7 +41,8 @@ LABEL_MAPPING_REVERSED = {v:k for k,v in LABEL_MAPPING.items() }
41
  #pipeline
42
  text_cls_pipeline = pipeline(task='sentiment-analysis',
43
  tokenizer=tokenizer,
44
- model=model)
 
45
 
46
  css_text = """<link rel="stylesheet" href="https://www.w3schools.com/w3css/4/w3.css">"""
47
  def render_html(items: List[Dict]):
 
41
  #pipeline
42
  text_cls_pipeline = pipeline(task='sentiment-analysis',
43
  tokenizer=tokenizer,
44
+ model=model,
45
+ return_all_scores=True)
46
 
47
  css_text = """<link rel="stylesheet" href="https://www.w3schools.com/w3css/4/w3.css">"""
48
  def render_html(items: List[Dict]):