more fixes

app.py

@@ -1,10 +1,14 @@
 import gradio as gr
 import re
+import os
 from mailparser import parse_from_string
 from bs4 import BeautifulSoup
+from gliner import GLiNER
 import spacy
 
 nlp = spacy.load("en_core_web_sm")
+_MODEL = {}
+_CACHE_DIR = os.environ.get("CACHE_DIR", None)
 
 def accept_mail(email_content):
     email = parse_from_string(email_content)
@@ -33,6 +37,17 @@ def get_sentences(further_cleaned_text):
     #         entities.append((ent.text, ent.label_))
     # return entities
 
+def get_model(model_name: str = None):
+    if model_name is None:
+        model_name = "urchade/gliner_base"
+
+    global _MODEL
+
+    if _MODEL.get(model_name) is None:
+        _MODEL[model_name] = GLiNER.from_pretrained(model_name, cache_dir=_CACHE_DIR)
+
+    return _MODEL[model_name]
+
 def present(email_content, labels):
     email = accept_mail(email_content)
     cleaned_text = clean_email(email)

requirements.txt

@@ -2,5 +2,5 @@ gliner
 mail-parser
 gradio
 beautifulsoup4
-spacy>=3.0
+spacy>=3.1.3
 https://github.com/explosion/spacy-models/releases/download/en_core_web_sm-3.0.0/en_core_web_sm-3.0.0.tar.gz

spaces.yml

@@ -1,4 +1,4 @@
 # spaces.yml
 pip:
-  - spacy>=3.0
-  - https://github.com/explosion/spacy-models/releases/download/en_core_web_sm-3.0.0/en_core_web_sm-3.0.0.tar.gz
+  - spacy==3.1.3
+  - https://github.com/explosion/spacy-models/releases/download/en_core_web_sm-3.1.0/en_core_web_sm-3.1.0.tar.gz