Working locally

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 from transformers import pipeline
+import json
 
 def process_swedish_text(text):
   # Models from https://huggingface.co/models
@@ -9,8 +10,11 @@ def process_swedish_text(text):
   # Run NER
   pipeline_results = ner_pipeline(text)
   print('NER results:', pipeline_results)
+  # Fix TypeError("'numpy.float32' object is not iterable")
+  pipeline_results_adjusted = map(lambda entity: entity | { 'score': float(entity['score']) }, pipeline_results)
+  print(pipeline_results_adjusted)
   # Return values
-  return pipeline_results
+  return json.dumps({'entities': list(pipeline_results_adjusted)})
 
 gradio_interface = gr.Interface(fn=process_swedish_text, inputs="text", outputs="json")
 gradio_interface.launch()

requirements.txt

@@ -1 +1,170 @@
-transformers==4.23.1
+aiohttp==3.8.3
+aiosignal==1.2.0
+anyio==3.6.2
+appnope==0.1.3
+argon2-cffi==21.3.0
+argon2-cffi-bindings==21.2.0
+asttokens==2.0.8
+async-timeout==4.0.2
+attrs==22.1.0
+Babel==2.10.3
+backcall==0.2.0
+bcrypt==4.0.1
+beautifulsoup4==4.11.1
+bleach==5.0.1
+blis==0.7.9
+catalogue==2.0.8
+certifi==2022.6.15
+cffi==1.15.1
+charset-normalizer==2.1.0
+click==8.1.3
+confection==0.0.3
+contourpy==1.0.5
+cryptography==38.0.4
+cycler==0.11.0
+cymem==2.0.7
+datasets==2.6.1
+debugpy==1.6.3
+decorator==5.1.1
+defusedxml==0.7.1
+dill==0.3.5.1
+entrypoints==0.4
+executing==1.1.1
+fastai==2.7.9
+fastapi==0.88.0
+fastbook==0.0.28
+fastcore==1.5.27
+fastdownload==0.0.7
+fastjsonschema==2.16.2
+fastprogress==1.0.3
+ffmpy==0.3.0
+filelock==3.8.0
+Flask==2.2.2
+fonttools==4.38.0
+frozenlist==1.3.1
+fsspec==2022.10.0
+gradio==3.12.0
+graphviz==0.20.1
+gunicorn==20.1.0
+h11==0.12.0
+httpcore==0.15.0
+httpx==0.23.1
+huggingface-hub==0.10.1
+idna==3.3
+importlib-metadata==5.0.0
+ipykernel==6.16.1
+ipython==8.5.0
+ipython-genutils==0.2.0
+ipywidgets==8.0.2
+itsdangerous==2.1.2
+jedi==0.18.1
+Jinja2==3.1.2
+joblib==1.2.0
+json5==0.9.10
+jsonschema==4.16.0
+jupyter-server==1.21.0
+jupyter_client==7.4.3
+jupyter_core==4.11.2
+jupyterlab-pygments==0.2.2
+jupyterlab-widgets==3.0.3
+jupyterlab_server==2.16.1
+kiwisolver==1.4.4
+langcodes==3.3.0
+linkify-it-py==1.0.3
+markdown-it-py==2.1.0
+MarkupSafe==2.1.1
+matplotlib==3.6.1
+matplotlib-inline==0.1.6
+mdit-py-plugins==0.3.1
+mdurl==0.1.2
+mistune==2.0.4
+multidict==6.0.2
+multiprocess==0.70.13
+murmurhash==1.0.9
+nbclassic==0.4.5
+nbclient==0.7.0
+nbconvert==7.2.2
+nbformat==5.7.0
+nest-asyncio==1.5.6
+notebook==6.5.1
+notebook_shim==0.2.0
+numpy==1.23.4
+orjson==3.8.2
+packaging==21.3
+pandas==1.5.1
+pandocfilters==1.5.0
+paramiko==2.12.0
+parso==0.8.3
+pathy==0.6.2
+pexpect==4.8.0
+pickleshare==0.7.5
+Pillow==9.2.0
+preshed==3.0.8
+prometheus-client==0.15.0
+prompt-toolkit==3.0.31
+psutil==5.9.3
+psycopg2==2.9.5
+ptyprocess==0.7.0
+pure-eval==0.2.2
+pyarrow==9.0.0
+pycparser==2.21
+pycryptodome==3.16.0
+pydantic==1.10.2
+pydub==0.25.1
+Pygments==2.13.0
+PyNaCl==1.5.0
+pyparsing==3.0.9
+pyrsistent==0.18.1
+python-dateutil==2.8.2
+python-dotenv==0.20.0
+python-multipart==0.0.5
+pytz==2022.5
+PyYAML==6.0
+pyzmq==24.0.1
+regex==2022.9.13
+requests==2.28.1
+responses==0.18.0
+rfc3986==1.5.0
+scikit-learn==1.1.2
+scipy==1.9.3
+Send2Trash==1.8.0
+sentencepiece==0.1.97
+six==1.16.0
+smart-open==5.2.1
+sniffio==1.3.0
+soupsieve==2.3.2.post1
+spacy==3.4.2
+spacy-legacy==3.0.10
+spacy-loggers==1.0.3
+srsly==2.4.5
+stack-data==0.5.1
+starlette==0.22.0
+stripe==3.5.0
+terminado==0.16.0
+thinc==8.1.5
+threadpoolctl==3.1.0
+tinycss2==1.2.1
+tokenizers==0.13.1
+tomli==2.0.1
+torch==1.12.1
+torchvision==0.13.1
+tornado==6.2
+tqdm==4.64.1
+traitlets==5.5.0
+transformers==4.23.1
+typer==0.4.2
+typing_extensions==4.4.0
+uc-micro-py==1.0.1
+urllib3==1.26.10
+uvicorn==0.20.0
+wasabi==0.10.1
+wcwidth==0.2.5
+webencodings==0.5.1
+websocket-client==1.4.1
+websockets==10.4
+Werkzeug==2.2.2
+widgetsnbextension==4.0.3
+XlsxWriter==3.0.3
+xxhash==3.1.0
+yarl==1.8.1
+zipp==3.9.0