Update app.py

app.py

@@ -5,14 +5,14 @@ import torch, json
 
 token = AutoTokenizer.from_pretrained('distilbert-base-uncased-finetuned-sst-2-english')
 
-types = [0,1]
+types = [{0:'positive'},{1:'Negative'}]
 
 inf_session = onnxruntime.InferenceSession('classifier_quantized.onnx')
 input_name = inf_session.get_inputs()[0].name
 output_name = inf_session.get_outputs()[0].name
 
 def classify(review):
-    input_ids = token(review)['inputs_ids'][:512]
+    input_ids = token(review)['input_ids'][:512]
     logits = inf_session.run([output_name],{input_name: [input_ids]})[0]
     logits = torch.FloatTensorlogits(logits)
     probs = torch.sigmoid(logits)[0]