Update app.py

app.py

@@ -14,7 +14,7 @@ output_name = inf_session.get_outputs()[0].name
 def classify(review):
     input_ids = token(review)['input_ids'][:512]
     logits = inf_session.run([output_name],{input_name: [input_ids]})[0]
-    logits = torch.FloatTensorlogits(logits)
+    logits = torch.FloatTensor(logits)
     probs = torch.sigmoid(logits)[0]
     return dict(zip(types,map(float,probs)))