Rimi98 commited on
Commit
16da45f
1 Parent(s): ed56b9b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import torch, json
5
 
6
  token = AutoTokenizer.from_pretrained('distilroberta-base')
7
 
8
- types = ['Toxic','Severe_toxic','Obscene','Threat','Insult','Identity_hate','Positive']
9
 
10
  inf_session = onnxruntime.InferenceSession('classifier-quantized.onnx')
11
  input_name = inf_session.get_inputs()[0].name
 
5
 
6
  token = AutoTokenizer.from_pretrained('distilroberta-base')
7
 
8
+ types = ['Toxic','Severe_toxic','Obscene','Threat','Insult','Identity_hate']
9
 
10
  inf_session = onnxruntime.InferenceSession('classifier-quantized.onnx')
11
  input_name = inf_session.get_inputs()[0].name