Update app.py

app.py

@@ -1,13 +1,15 @@
 import gradio as gr
 from sentence_transformers import SentenceTransformer, util
 import string, re
+from cleanco import basename
 
 model = None
 def prepare(text):
     text = text.translate(str.maketrans('', '', string.punctuation + '”“'))
-    pattern = r"\b(?=[MDCLXVIΙ])M{0,4}(CM|CD|D?C{0,3})(XC|XL|L?X{0,3})([IΙ]X|[IΙ]V|V?[IΙ]{0,3})\b\.?"
+    pattern = r"\b(?=[MDCLXVII])M{0,4}(CM|CD|D?C{0,3})(XC|XL|L?X{0,3})([II]X|[II]V|V?[II]{0,3})\b\.?"
     text = re.sub(pattern, '', text)
-    return text.upper()
+    text = basename(text).upper()
+    return text
     
 def semantic(company_1, company_2):
     global model