Rimi98 commited on
Commit
88c246f
1 Parent(s): 3c11d30

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ input_name = inf_session.get_inputs()[0].name
12
  output_name = inf_session.get_outputs()[0].name
13
 
14
  def classify(review):
15
- input_ids = token(description)['input_ids'][:512]
16
  logits = inf_session.run([output_name], {input_name: [input_ids]})[0]
17
  logits = torch.FloatTensor(logits)
18
  probs = torch.sigmoid(logits)[0]
 
12
  output_name = inf_session.get_outputs()[0].name
13
 
14
  def classify(review):
15
+ input_ids = token(review)['input_ids'][:512]
16
  logits = inf_session.run([output_name], {input_name: [input_ids]})[0]
17
  logits = torch.FloatTensor(logits)
18
  probs = torch.sigmoid(logits)[0]