Update app.py

app.py

@@ -29,6 +29,8 @@ def process_output_trad(pred):
     return pred.split()
 
 def get_id_picto_from_predicted_lemma(df_lexicon, lemma):
+    if lemma.endswith("!"):
+        lemma = lemma[:-1]
     id_picto = df_lexicon.loc[df_lexicon['keyword_no_cat'] == lemma, 'id_picto'].tolist()
     return (id_picto[0], lemma) if id_picto else (0, lemma)