Add Application files

app.py

@@ -1,4 +1,36 @@
-import streamlit as st
+import gradio
+from transformers import pipline
 
-x = st.slider("Select a value:")
-st.write(x, "square is ", x * x)
+def merge_split_token(tokens):
+  merged = []
+  for token in tokens:
+    if token["word"].startswith('##'):
+      merged[-1]["word"] += token["word"][2:]
+    else:
+      merged.append(token)
+
+  return merged
+def process_trans_text(text):
+  nlp=pipline("ner",  model='KBLab/bert-base-swedish-cased-ner', tokenizer='KBLab/bert-base-swedish-cased-ner')
+  nlp_results = nlp(text)
+  print('nlp_results:', nlp_results)
+
+  nlp_results_merge = merge_split_token(nlp_results)
+  nlp_results_adjusted = map(lambda entity: dict(entity, **{ 'score': float(entity['score']) }), nlp_results_merge)
+  print('nlp_results_adjusted:', nlp_results_adjusted)
+  # Return values
+  return {'entities': list(nlp_results_adjusted)}
+
+gradio_intreface = gradio.Interface(
+  fn=process_trans_text,
+  inputs="text",
+  output="text",
+  examples=[
+    ["Jag heter Tom och bor i Stockholm."],
+    ["Groens malmgård är en av Stockholms malmgårdar, belägen vid Malmgårdsvägen 53 på Södermalm i Stockholm."]
+  ],
+  title="Entity Recognition",
+  description="Something text" 
+)
+
+gradio_intreface.launch()

requirements.txt

@@ -1,44 +1,69 @@
+aiofiles==23.2.1
 altair==5.2.0
+annotated-types==0.6.0
+anyio==4.2.0
 attrs==23.2.0
-blinker==1.7.0
-cachetools==5.3.2
 certifi==2023.11.17
 charset-normalizer==3.3.2
 click==8.1.7
-gitdb==4.0.11
-GitPython==3.1.41
+colorama==0.4.6
+contourpy==1.2.0
+cycler==0.12.1
+fastapi==0.109.0
+ffmpy==0.3.1
+filelock==3.13.1
+fonttools==4.47.2
+fsspec==2023.12.2
+gradio==4.15.0
+gradio_client==0.8.1
+h11==0.14.0
+httpcore==1.0.2
+httpx==0.26.0
+huggingface-hub==0.20.3
 idna==3.6
-importlib-metadata==7.0.1
+importlib-resources==6.1.1
 Jinja2==3.1.3
 jsonschema==4.21.1
 jsonschema-specifications==2023.12.1
+kiwisolver==1.4.5
 markdown-it-py==3.0.0
 MarkupSafe==2.1.4
+matplotlib==3.8.2
 mdurl==0.1.2
 numpy==1.26.3
+orjson==3.9.12
 packaging==23.2
 pandas==2.2.0
 pillow==10.2.0
-protobuf==4.25.2
-pyarrow==15.0.0
-pydeck==0.8.1b0
+pydantic==2.5.3
+pydantic_core==2.14.6
+pydub==0.25.1
 Pygments==2.17.2
+pyparsing==3.1.1
 python-dateutil==2.8.2
+python-multipart==0.0.6
 pytz==2023.3.post1
+PyYAML==6.0.1
 referencing==0.32.1
+regex==2023.12.25
 requests==2.31.0
 rich==13.7.0
 rpds-py==0.17.1
+ruff==0.1.14
+safetensors==0.4.2
+semantic-version==2.10.0
+shellingham==1.5.4
 six==1.16.0
-smmap==5.0.1
-streamlit==1.30.0
-tenacity==8.2.3
-toml==0.10.2
+sniffio==1.3.0
+starlette==0.35.1
+tokenizers==0.15.1
+tomlkit==0.12.0
 toolz==0.12.1
-tornado==6.4
+tqdm==4.66.1
+transformers==4.37.0
+typer==0.9.0
 typing_extensions==4.9.0
 tzdata==2023.4
-tzlocal==5.2
 urllib3==2.1.0
-validators==0.22.0
-zipp==3.17.0
+uvicorn==0.27.0
+websockets==11.0.3