thak123 commited on
Commit
385285d
1 Parent(s): 80266c4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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
- pd.DataFrame({"original_predictions":predictions}).to_csv("eacl_slavic.tsv")
 
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(