| import gradio as gr | |
| from transformers import T5Tokenizer, T5ForConditionalGeneration | |
| tokenizer = T5Tokenizer.from_pretrained("./model_files") | |
| model = T5ForConditionalGeneration.from_pretrained("./model_files") | |
| def greet(name): | |
| inp = "Summarize:" + name | |
| input_ids = tokenizer(inp, return_tensors="pt").input_ids | |
| #return tokenizer.decode(outputs[0],skip_special_tokens=True, clean_up_tokenization_spaces=True) | |
| return inp | |
| # We instantiate the Textbox class | |
| textbox = gr.Textbox(label="Type your name here:", placeholder="Article", lines=5) | |
| gr.Interface(fn=greet, inputs=textbox, outputs="text").launch() |