Rimi98 commited on
Commit
7497866
·
1 Parent(s): 66b7aa9

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_quantizer.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:'Positive'},{1:'Negative'}]
9
 
10
  inf_session = onnxruntime.InferenceSession('classifier_quantizer.onnx')
11
  input_name = inf_session.get_inputs()[0].name