Update README.md

README.md

@@ -7,7 +7,7 @@ widget:
 ---
 
 # Spanish-T5-small fine-tuned on **SQAC** for QA 📖❓
-[Google's mT5-small](https://huggingface.co/flax-community/spanish-t5-small) fine-tuned on [SQAC](https://huggingface.co/datasets/BSC-TeMU/SQAC) (secondary task) for **Q&A** downstream task.
+[spanish-T5-small](https://huggingface.co/flax-community/spanish-t5-small) fine-tuned on [SQAC](https://huggingface.co/datasets/BSC-TeMU/SQAC) (secondary task) for **Q&A** downstream task.
 
 ## Details of Spanish T5 (small)
 
@@ -24,7 +24,7 @@ widget:
 
 | Metric | # Value   |
 | ------ | --------- |
-| **EM** | **41.65** |
+| **BLEU** | **41.94** |
 
 
 
@@ -34,8 +34,9 @@ widget:
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-tokenizer = AutoTokenizer.from_pretrained("mrm8488/mT5-small-finetuned-tydiqa-for-xqa")
-model = AutoModelForCausalLM.from_pretrained("mrm8488/mT5-small-finetuned-tydiqa-for-xqa").to(device)
+ckpt = 'spanish-t5-small-sqac-for-qa'
+tokenizer = AutoTokenizer.from_pretrained(ckpt)
+model = AutoModelForCausalLM.from_pretrained(ckpt).to(device)
 
 def get_response(question, context, max_length=32):
   input_text = 'question: %s  context: %s' % (question, context)
@@ -47,19 +48,6 @@ def get_response(question, context, max_length=32):
 
   return tokenizer.decode(output[0], skip_special_tokens=True)
   
-# Some examples in different languages
-
-context = 'HuggingFace won the best Demo paper at EMNLP2020.'
-question = 'What won HuggingFace?'
-get_response(question, context)
-
-context = 'HuggingFace ganó la mejor demostración con su paper en la EMNLP2020.'
-question = 'Qué ganó HuggingFace?'
-get_response(question, context)
-
-context = 'HuggingFace выиграл лучшую демонстрационную работу на EMNLP2020.'
-question = 'Что победило в HuggingFace?'
-get_response(question, context)
 ```
 
 > Created by [Manuel Romero/@mrm8488](https://twitter.com/mrm8488) | [LinkedIn](https://www.linkedin.com/in/manuel-romero-cs/)