Spaces:
Runtime error
Runtime error
| import gradio as gr | |
| from simplet5 import SimpleT5 | |
| model = SimpleT5() | |
| model.from_pretrained(model_type="t5", model_name="t5-base") | |
| def summarize(text_input): | |
| if text_input == None: | |
| return None | |
| model.load_model("data.pkl", use_gpu=False) | |
| text_to_summarize = """summarize: """ + text_input | |
| predicted_output = model.predict(text_to_summarize) | |
| return predicted_output | |
| text_i = gr.inputs.Textbox(lines=10, placeholder="Enter text", label="Text") | |
| iface = gr.Interface(fn = summarize, inputs = text_i, outputs="textbox") | |
| iface.launch() |