Fixing result error

app.py

@@ -21,8 +21,8 @@ def predict(text):
 
     outputs = model(**inputs)
     predictions = outputs.logits
-    predictionItem = predictions.item()
-    result = torch.sum(predictionItem)
+    # predictionItem = predictions.item()
+    result = torch.sum(predictions)
     result = max(result, 0)
     result = min(result, 5)
     return mapping[round(result)]