IlyaGusev
/

mbart_ru_sum_gazeta

text2text-generation

Inference Endpoints

Model card Files Files and versions Community

IlyaGusev commited on Sep 3, 2021

Commit

310a4d4

·

1 Parent(s): d5f1442

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -98,7 +98,7 @@ def predict(
         for line in r:
             record = json.loads(line)
             inputs.append(record["text"])
-            targets.append(record["summary"])
     tokenizer = MBartTokenizer.from_pretrained(model_name)
     device = torch.device("cuda:0") if use_cuda else torch.device("cpu")

         for line in r:
             record = json.loads(line)
             inputs.append(record["text"])
+            targets.append(record["summary"].replace("\n", " "))
     tokenizer = MBartTokenizer.from_pretrained(model_name)
     device = torch.device("cuda:0") if use_cuda else torch.device("cpu")