Spaces:
Sleeping
Sleeping
| import time | |
| import gradio as gr | |
| from litie.pipelines import EventExtractionPipeline | |
| pipeline = EventExtractionPipeline(task_model_name="gplinker", model_name_or_path="xusenlin/duee-gplinker") | |
| def extract(text): | |
| start = time.time() | |
| res = pipeline(text) | |
| running_time = time.time() - start | |
| return running_time, res | |
| demo = gr.Interface( | |
| extract, | |
| [ | |
| gr.Textbox( | |
| placeholder="Enter sentence here...", | |
| lines=5 | |
| ), | |
| ], | |
| [gr.Number(label="Run Time"), gr.Json(label="Result")], | |
| examples=[ | |
| ["油服巨头哈里伯顿裁员650人 因美国油气开采活动放缓"], | |
| ], | |
| title="Event Extraction Demo", | |
| ) | |
| demo.launch() | |