Update app.py

app.py

@@ -16,7 +16,7 @@ def classify(review):
     input_ids = token(review)['input_ids'][:512]
     logits = inf_session.run([output_name],{input_name : [input_ids]})[0]
     logits = torch.FloatTensor(logits)
-    prob = torch.sigmoid(logits)[0]
+    probs = torch.sigmoid(logits)[0]
     x = 2
     return dict(zip(classes,map(float,probs)))