Rimi98 commited on
Commit
f688dec
·
1 Parent(s): ef9cff7

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('distilbert-base-uncased-finetuned-sst-2-english')
7
 
8
- types = [{0:'Positive'},{1:'Negative'}]
9
 
10
  inf_session = onnxruntime.InferenceSession('classifier_quantized.onnx')
11
  input_name = inf_session.get_inputs()[0].name
 
5
 
6
  token = AutoTokenizer.from_pretrained('distilbert-base-uncased-finetuned-sst-2-english')
7
 
8
+ types = [0,1]
9
 
10
  inf_session = onnxruntime.InferenceSession('classifier_quantized.onnx')
11
  input_name = inf_session.get_inputs()[0].name