Create README.md

README.md

@@ -0,0 +1,32 @@
+---
+language: ["es"]
+tags:
+- spanish
+- question generation
+- qg
+Datasets:
+- SQUAD
+license: mit
+---
+This is the finetuned model of hiiamsid/est5-base for Question Generation task.
+* Here input is the context only and output is questions. No information regarding answers were given to model. 
+* Unfortunately, due to lack of sufficient resources it is fine tuned with batch_size=10 and num_seq_len=256. So, if too large context is given model may not get information about last portions.
+
+```
+from transformers import T5ForConditionalGeneration, T5Tokenizer
+MODEL_NAME = 'hiiamsid/est5-base-qg'
+model = T5ForConditionalGeneration.from_pretrained(MODEL_NAME)
+tokenizer = T5Tokenizer.from_pretrained(MODEL_NAME)
+model.cuda();
+model.eval();
+def generate_question(text, beams=10, grams=2, num_return_seq=10,max_size=256):
+    x = tokenizer(text, return_tensors='pt', padding=True).to(model.device)
+    out = model.generate(**x, no_repeat_ngram_size=grams, num_beams=beams, num_return_sequences=num_return_seq, max_length=max_size)
+    return tokenizer.decode(out[0], skip_special_tokens=True)
+print(generate_question('Any context in spanish from which question is to be generated'))
+
+```
+
+## Citing & Authors
+- Datasets : [squad_es](https://huggingface.co/datasets/squad_es)
+- Model : [hiiamsid/est5-base](hiiamsid/est5-base)