Jaehan/Text2Text-Question-Generation-1

1from transformers import AutoModelWithLMHead, AutoTokenizer2import gradio as gr3 4model_name = "mrm8488/t5-base-finetuned-question-generation-ap"5text2text_tokenizer = AutoTokenizer.from_pretrained(model_name)6model = AutoModelWithLMHead.from_pretrained(model_name)7 8def text2text(context, answer):9    input_text = f"answer: {answer}  context: <{context}>"10    features = text2text_tokenizer([input_text], return_tensors="pt")11    output = model.generate(input_ids=features["input_ids"], attention_mask=features["attention_mask"], max_length=100)12    response = text2text_tokenizer.decode(output[0])13    return response14 15context = gr.Textbox(lines=10, label="English", placeholder="Context")16answer = gr.Textbox(lines=1, label="Answer")17out = gr.Textbox(lines=1, label="Generated question")18 19gr.Interface(text2text, inputs=[context, answer], outputs=out).launch()