Update app.py

app.py

@@ -5,8 +5,9 @@ dicta = Dicta('./dicta-1.0.int8.onnx')
 
 theme = gr.themes.Soft(font=[gr.themes.GoogleFont("Roboto")])
 
-def add_diacritics(text: str) -> str:
-    return dicta.add_diacritics(text)
+def add_diacritics(text: str, mark_niqqud_male: bool) -> str:
+    mark_matres_lectionis = "|" if mark_niqqud_male else None
+    return dicta.add_diacritics(text, mark_matres_lectionis=mark_matres_lectionis)
 
 with gr.Blocks(theme=theme) as demo:
     gr.Markdown("# Hebrew Text Diacritization")
@@ -19,6 +20,9 @@ with gr.Blocks(theme=theme) as demo:
         rtl=True,
         elem_classes=["input"]
     )
+
+    mark_niqqud_male = gr.Checkbox(label="Mark Niqqud Male (Matres Lectionis)")
+
     diacritized_output = gr.Textbox(
         label="Text with Diacritics",
         lines=7,
@@ -30,7 +34,7 @@ with gr.Blocks(theme=theme) as demo:
 
     submit_button.click(
         fn=add_diacritics,
-        inputs=[text_input],
+        inputs=[text_input, mark_niqqud_male],
         outputs=[diacritized_output]
     )