jeffhaines commited on
Commit
54235b5
·
1 Parent(s): ab0c546

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('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 (https://github.com/hendrycks/ethics). It judges whether a given action of scenario is wrong or not wrong and shows how the words in the scenario affected the judgment.')
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 transformer-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'