Change to show all prediction scores instead of just one of the predicted class

#1
by winniehcy - opened
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ def strip_input_str(x):
16
  return x.strip()
17
 
18
  # Load the pipeline with the HanmunRoBERTa model
19
- model_pipeline = pipeline(task="text-classification", model="bdsl/HanmunRoBERTa")
20
 
21
  # Streamlit app layout
22
  title = "HanmunRoBERTa Century Classifier"
 
16
  return x.strip()
17
 
18
  # Load the pipeline with the HanmunRoBERTa model
19
+ model_pipeline = pipeline(task="text-classification", model="bdsl/HanmunRoBERTa", return_all_scores=True)
20
 
21
  # Streamlit app layout
22
  title = "HanmunRoBERTa Century Classifier"