rounding

app.py

@@ -36,8 +36,9 @@ def predict (Sentence, Word, LLM, Norm, Layer):
     data = (Sentence, Word)
     emb = lm.extract_representation(data, layer=Layer)
     pred = torch.nn.functional.relu(model(emb))
-    pred = pred.squeeze(0)
-    pred_list = pred.detach().numpy().tolist()
+    pred_sq = pred.squeeze(0)
+    pred_round = torch.round(pred_sq, decimals=2)
+    pred_list = pred_round.detach().numpy().tolist()
 
     df = pd.DataFrame({'feature':labels, 'value':pred_list})
     df = df[df['value'] > 0]