Update app.py

app.py

@@ -1,11 +1,17 @@
 import gradio as gr
 from sentence_transformers import SentenceTransformer, util
-
+import string
 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\.?"
+    text = re.sub(pattern, '', text)
+    return text.upper()
+    
 def semantic(company_1, company_2):
     global model 
     # Single list of sentences
-    sentences = [company_1, company_2]
+    sentences = [prepare(company_1), prepare(company_2)]
     if model is None:
         model = SentenceTransformer('all-mpnet-base-v2')
     #Compute embeddings