Commit
·
a39dc59
1
Parent(s):
54235b5
Update app.py
Browse files
app.py
CHANGED
@@ -7,7 +7,7 @@ from transformers_interpret import SequenceClassificationExplainer
|
|
7 |
from transformers import AutoModelForSequenceClassification, AutoTokenizer
|
8 |
|
9 |
st.title('Ethics Classifier')
|
10 |
-
st.write('This app uses a pre-trained Distilbert model fine-tuned on the Commonsense Ethics dataset from the Aligning AI With Shared Human Values project (for more information, see https://github.com/hendrycks/ethics). It judges whether a given action of scenario is wrong or not wrong and uses
|
11 |
|
12 |
loaded_model = DistilBertForSequenceClassification.from_pretrained('commonsense_ethics')
|
13 |
model_name = 'distilbert-base-uncased'
|
|
|
7 |
from transformers import AutoModelForSequenceClassification, AutoTokenizer
|
8 |
|
9 |
st.title('Ethics Classifier')
|
10 |
+
st.write('This app uses a pre-trained Distilbert model fine-tuned on the Commonsense Ethics dataset from the Aligning AI With Shared Human Values project (for more information, see https://github.com/hendrycks/ethics). It judges whether a given action of scenario is wrong or not wrong and uses transformers-interpret (https://pypi.org/project/transformers-interpret/) to show how the words in your scenario affected the model\'s judgment.')
|
11 |
|
12 |
loaded_model = DistilBertForSequenceClassification.from_pretrained('commonsense_ethics')
|
13 |
model_name = 'distilbert-base-uncased'
|