Hugging Face's logo

Spaces:
Rimi98
/
ReviewClassifier
Sleeping

App Files Community
Rimi98 commited on
Commit
cf9c79c
·
1 Parent(s): 0b3097e

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