Narrativa commited on
Commit
7e36559
1 Parent(s): 0f9b4f9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ classifier = pipeline("text-classification", model=model, tokenizer=tokenizer)
14
 
15
  def prediction(header, text):
16
  results = classifier(header + " [SEP] " + text)
17
- return results[0]["label"], round(results[0]["score"], 5)
18
 
19
 
20
  gradio_ui = gr.Interface(
 
14
 
15
  def prediction(header, text):
16
  results = classifier(header + " [SEP] " + text)
17
+ return results[0]["label"], results[0]["score"]
18
 
19
 
20
  gradio_ui = gr.Interface(