Update app.py

app.py

@@ -3,16 +3,36 @@ from dicta_onnx import Dicta
 
 dicta = Dicta('./dicta-1.0.int8.onnx')
 
-def add_diacritics(text):
+theme = gr.themes.Soft(font=[gr.themes.GoogleFont("Roboto")])
+
+def add_diacritics(text: str) -> str:
     return dicta.add_diacritics(text)
 
-iface = gr.Interface(
-    fn=add_diacritics,
-    inputs=gr.Textbox(placeholder='Type text without diacritics', lines=3, rtl=True),
-    outputs=gr.Textbox(label='Text with diacritics', rtl=True),
-    title='Add diacritics to Hebrew text',
-    description='Type and press the button to add diacritics.',
-    live=False,
-)
+with gr.Blocks(theme=theme) as demo:
+    gr.Markdown("# Hebrew Text Diacritization")
+    gr.Markdown("Enter Hebrew text without diacritics, and the model will add them.")
+
+    text_input = gr.Textbox(
+        placeholder="Type text without diacritics...",
+        label="Input Text",
+        lines=5,
+        rtl=True,
+        elem_classes=["input"]
+    )
+    diacritized_output = gr.Textbox(
+        label="Text with Diacritics",
+        lines=7,
+        rtl=True,
+        elem_classes=["output"]
+    )
+    
+    submit_button = gr.Button("Add Diacritics")
+
+    submit_button.click(
+        fn=add_diacritics,
+        inputs=[text_input],
+        outputs=[diacritized_output]
+    )
 
-iface.launch()
+if __name__ == "__main__":
+    demo.launch()