Spaces:

dayannex
/

anonimizador_V2

Sleeping

dayannex commited on Aug 22, 2024

Commit

58bcaa2

1 Parent(s): 6ca7324

dataset model csv ingles

Files changed (1) hide show

app.py CHANGED Viewed

@@ -486,7 +486,7 @@ class ModeloDataset:
             logits = outputs.logits
             predictions = torch.argmax(logits, dim=2)
-            id2label = model.config.id2label
             all_tokens = []

             logits = outputs.logits
             predictions = torch.argmax(logits, dim=2)
+            id2label = self.model.config.id2label
             all_tokens = []