Spaces:
Runtime error
Runtime error
| import gradio as gr | |
| # Models from https://huggingface.co/models | |
| # https://huggingface.co/KBLab/bert-base-swedish-cased-ner | |
| ml_model = 'KBLab/bert-base-swedish-cased-ner' | |
| ner_pipeline = pipeline(model=ml_model, task='ner') | |
| def process_swedish_text(text): | |
| pipeline_results = ner_pipeline(text) | |
| print('NER results:', pipeline_results) | |
| pipeline_results_adjusted = map(lambda entity: entity | { 'score': float(entity['score']) }, pipeline_results) | |
| print(pipeline_results_adjusted) | |
| return pipeline_results_adjusted | |
| gradio_interface = gr.Interface(fn=process_swedish_text, inputs="text", outputs="JSON") | |
| gradio_interface.launch() | |