some imrovemnts

app.py

@@ -7,7 +7,11 @@ nlp = spacy.load("sr_Spacy_Serbian_Model_SrpKor4Tagging_BERTICOVO")
 
 # Define a function to display the tags and lemmas
 def display_tags_and_lemmas(text):
-    doc = nlp(text)
+    # First, we'll use spaCy to tag and parse the text
+    text = text.strip()
+    if not text:
+        return "", ""
+    doc = nlp(text.strip())
     html = displacy.render(doc, style="ent", page=True, minify=True)
 
     # We'll also create a custom HTML to display lemmas nicely
@@ -29,7 +33,8 @@ iface = gr.Interface(
     outputs=gr.HTML(label="Leme i POS oznake"),
     title="spaCy Tagger i Lemmatizer",
     description="Unesite rečenicu da biste videli njene imenovane entitete, POS oznake i leme.",
-    examples=["Lep je dan, danas. Sutra će biti još lepši!", "Psi su trčali svakog dana. Mačke su spavale."]
+    examples=["Lep je dan, danas. Sutra će biti još lepši!", "Psi su trčali svakog dana. Mačke su spavale."],
+    allow_screenshot=True
 )
 
 iface.launch()