Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -70,7 +70,8 @@ for _, batch in enumerate(features_dict["document"]['test']):
|
|
70 |
prediction =torch.max(classifier_output.logits, axis=1)
|
71 |
predictions.append(prediction.indices.item())
|
72 |
|
73 |
-
|
|
|
74 |
|
75 |
|
76 |
trainer = MultitaskTrainer(
|
|
|
70 |
prediction =torch.max(classifier_output.logits, axis=1)
|
71 |
predictions.append(prediction.indices.item())
|
72 |
|
73 |
+
print("p:", predictions)
|
74 |
+
# pd.DataFrame({"original_predictions":predictions}).to_csv("eacl_slavic.tsv")
|
75 |
|
76 |
|
77 |
trainer = MultitaskTrainer(
|