Refactor app.py to utilize GPU allocation for NLP pipelines and update documentation for clarity and completeness

app.py

@@ -1,13 +1,19 @@
+import spaces  # isort:skip
 import gradio as gr
 from gr_nlp_toolkit import Pipeline
 
 # Author: Lefteris Loukas
-# Date: August 2024
-# Description: A Gradio interface for the Grεεk NLP Toolkit, which includes Greeklish to Greek conversion, dependency parsing, part-of-speech tagging, and named entity recognition.
+# Date: January 2025
+# Description: A Gradio interface for the Greek NLP Toolkit (gr-nlp-toolkit), which includes Greeklish to Greek conversion, dependency parsing, part-of-speech tagging, and named entity recognition.
 # Point-of-Contact: http://nlp.cs.aueb.gr/
 
+
 # Initialize Pipelines
-nlp_pos_ner_dp_with_g2g = Pipeline("pos,ner,dp,g2g")
+@spaces.GPU
+def allocate_pipeline():
+    nlp_pipeline = Pipeline("pos,ner,dp,g2g")
+    return nlp_pipeline
+
 
 G2G_PLACEHOLDER = "e.g., H thessaloniki einai mia poli sti boreia ellada"
 NER_PLACEHOLDER = "e.g., Η Αργεντινή κέρδισε το Παγκόσμιο Κύπελλο το 2022"
@@ -15,16 +21,22 @@ POS_PLACEHOLDER = "e.g., Μου αρέσει να διαβάζω τα post το
 DP_PLACEHOLDER = "e.g., Προτιμώ την πρωινή πτήση από την Αθήνα στη Θεσσαλονίκη."
 
 
+@spaces.GPU
 def greeklish_to_greek(text):
     if not text:
         text = G2G_PLACEHOLDER[5:]
 
-    doc = nlp_pos_ner_dp_with_g2g(text)
+    # doc = nlp_pos_ner_dp_with_g2g(text)
+    nlp_pipeline = allocate_pipeline()
+    doc = nlp_pipeline(text)
     return " ".join([token.text for token in doc.tokens])
 
 
+@spaces.GPU
 def process_text(text, task):
-    doc = nlp_pos_ner_dp_with_g2g(text)
+    # doc = nlp_pos_ner_dp_with_g2g(text)
+    nlp_pipeline = allocate_pipeline()
+    doc = nlp_pipeline(text)
     task_mapping = {
         "dp": lambda token: f"Text: {token.text}, Head: {token.head}, Deprel: {token.deprel}",
         "pos": lambda token: f"Text: {token.text}, UPOS: {token.upos}, Feats: {token.feats}",
@@ -58,8 +70,15 @@ def create_demo():
     with gr.Blocks(theme=theme) as demo:
         gr.Markdown(
             """
-        # The Greek NLP Toolkit 🇬🇷
-        This is a demonstration space for our open-source Python toolkit (`gr-nlp-toolkit`), which supports state-of-the-art natural language processing capabilities in Greek. 
+        # GR-NLP-TOOLKIT Playground 🇬🇷
+
+        <p align="left">
+            <a href="https://github.com/nlpaueb/gr-nlp-toolkit">
+                <img src="https://github.com/nlpaueb/gr-nlp-toolkit/blob/main/logo.png?raw=true" width="200">
+            </a>
+        </p>
+        
+        This is an interactive playground/demo for our open-source Python toolkit (`gr-nlp-toolkit`), which supports state-of-the-art natural language processing capabilities in Greek. 
         
         ## Key Features:
         - Named Entity Recognition (NER)
@@ -122,13 +141,28 @@ def create_demo():
 
         ## Github Repository
 
-        Visit the [GitHub repository]("https://github.com/nlpaueb/gr-nlp-toolkit") for more information, such as documentation and full usage examples.
+        Visit the <a href="https://github.com/nlpaueb/gr-nlp-toolkit" target="_blank">GitHub repository</a> for more information, such as documentation and full usage examples.
+
+        ## Paper
+        The software was presented at COLING 2025. Read the full technical report/paper here: https://arxiv.org/abs/2412.08520
+
+        If you use our toolkit, please cite it:
+        ```bibtex
+        @misc{loukas2024grnlptoolkitopensourcenlptoolkit,
+            title={GR-NLP-TOOLKIT: An Open-Source NLP Toolkit for Modern Greek}, 
+            author={Lefteris Loukas and Nikolaos Smyrnioudis and Chrysa Dikonomaki and Spyros Barbakos and Anastasios Toumazatos and John Koutsikakis and Manolis Kyriakakis and Mary Georgiou and Stavros Vassos and John Pavlopoulos and Ion Androutsopoulos},
+            year={2024},
+            eprint={2412.08520},
+            archivePrefix={arXiv},
+            primaryClass={cs.CL},
+            url={https://arxiv.org/abs/2412.08520}, 
+        }
+        ```
 
         ## About the Project
         
-        [The Greek NLP Toolkit]("https://github.com/nlpaueb/gr-nlp-toolkit")  is the state-of-the-art natural language processing toolkit for modern Greek, [developed by the Natural Language Processing Group at the Athens University of Economics and Business](http://nlp.cs.aueb.gr/).
-        It supports named entity recognition, part-of-speech and morphological tagging, dependency parsing, 
-        and Greeklish to Greek conversion. This project is part of ongoing research aimed at advancing Greek language processing capabilities.
+        [The Greek NLP Toolkit](https://github.com/nlpaueb/gr-nlp-toolkit) is the state-of-the-art natural language processing toolkit for modern Greek, maintained by the <a href="http://nlp.cs.aueb.gr/" target="_blank">Natural Language Processing Group at the Athens University of Economics and Business</a>.
+        For technical questions, contact us via Github issues. For licensing and commercial inquiries, please contact us via the Contact page in the website.
         <br>
         <br>