jeffhaines commited on
Commit
51278b1
1 Parent(s): bc4b1cb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ from transformers_interpret import SequenceClassificationExplainer
7
  from transformers import AutoModelForSequenceClassification, AutoTokenizer
8
 
9
  st.title('Ethical Judgment 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('distilbert')
13
  model_name = 'distilbert-base-uncased'
7
  from transformers import AutoModelForSequenceClassification, AutoTokenizer
8
 
9
  st.title('Ethical Judgment Classifier')
10
+ st.write('This app uses a pre-trained Distilbert model fine-tuned on the Commonsense and Justice datasets 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('distilbert')
13
  model_name = 'distilbert-base-uncased'