Spaces:

OrganizedProgrammers
/

arXiv

Sleeping

Omar ID EL MOUMEN commited on Mar 27

Commit

d8045d1

1 Parent(s): 2e329bd

Change regex

Files changed (1) hide show

app.py CHANGED Viewed

@@ -104,7 +104,7 @@ async def extract_text_pdf(document: DocumentID):
         postprocess_text = remove_punctuations(postprocess_text)
         postprocess_text = re.sub(r"\s+", " ", postprocess_text)
         postprocess_text = postprocess_text.strip()
-        regex_titles = r"(?:[IVX]+|[0-9]+)\.\s[A-Z0-9\s]+"
         titles = doc.get_toc()
         main_titles = []
         if len(titles) <= 0:

         postprocess_text = remove_punctuations(postprocess_text)
         postprocess_text = re.sub(r"\s+", " ", postprocess_text)
         postprocess_text = postprocess_text.strip()
+        regex_titles = r"(?:[IVX]+|[0-9]+)\.\s[A-Z0-9\s]+$"
         titles = doc.get_toc()
         main_titles = []
         if len(titles) <= 0: