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

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 (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'
 
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'