Spaces:

snaramirez872
/

Finetuning-Toxicity-Model

Runtime error

snaramirez872 commited on Apr 26, 2023

Commit

ccbcc69

1 Parent(s): c018e35

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -158,5 +158,5 @@ def predict(tweets):
                 res.append({'TWEETS': tweets[i], 'LABEL': id2label[preds[i].item()], 'PROBABILITY': probs[i][preds[i]].item()})
     return res
-res = predict(testing_loader)
 st.table(res) # table

                 res.append({'TWEETS': tweets[i], 'LABEL': id2label[preds[i].item()], 'PROBABILITY': probs[i][preds[i]].item()})
     return res
+res = predict(test_loader_strings)
 st.table(res) # table