mrm8488
/

spanish-t5-small-sqac-for-qa

Text2Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

mrm8488 commited on Sep 3, 2021

Commit

61e89a3

·

1 Parent(s): 61d4da9

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -44,7 +44,7 @@ This dataset can be used to build extractive-QA.
 from transformers import T5ForConditionalGeneration, AutoTokenizer
 import torch
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-ckpt = 'spanish-t5-small-sqac-for-qa'
 tokenizer = AutoTokenizer.from_pretrained(ckpt)
 model = T5ForConditionalGeneration.from_pretrained(ckpt).to(device)

 from transformers import T5ForConditionalGeneration, AutoTokenizer
 import torch
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+ckpt = 'mrm8488/spanish-t5-small-sqac-for-qa'
 tokenizer = AutoTokenizer.from_pretrained(ckpt)
 model = T5ForConditionalGeneration.from_pretrained(ckpt).to(device)