round

app.py

@@ -97,7 +97,7 @@ def inference(model_name: str, sample: dict):
     tokens = merge_dialog_data(tokenizer, tokenized_dialog)
     with torch.inference_mode():
         logits = model(**tokens).logits
-        probas = torch.sigmoid(logits)[0].cpu().detach().numpy()
+        probas = torch.sigmoid(logits)[0].cpu().detach().numpy().tolist()
     return probas
 
 with st.spinner('Running inference...'):