Spaces:
Runtime error
Runtime error
Change to show all prediction scores instead of just one of the predicted class
#1
by
winniehcy
- opened
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"
|