Update app.py

app.py

@@ -47,11 +47,8 @@ download_file(url_eo, ckpt_path_eo)
 
 # ---------- Import required modules from the cloned repository ----------
 
-# Import the tokenizer and utility functions (shared by both models)
 from tashkeel_tokenizer import TashkeelTokenizer
 from utils import remove_non_arabic
-
-# Import the two model implementations under different names
 from ed_pl import TashkeelModel as TashkeelModel_ED
 from eo_pl import TashkeelModel as TashkeelModel_EO
 
@@ -103,7 +100,7 @@ def diacritize_text(model_type, input_text):
     text_clean = remove_non_arabic(input_text)
     x = [text_clean]
 
-    batch_size = 16  # Using a fixed batch size; adjust if necessary.
+    batch_size = 16  # Fixed batch size.
     verbose = False
 
     try:
@@ -116,7 +113,6 @@ def diacritize_text(model_type, input_text):
     except Exception as e:
         return f"An error occurred during diacritization: {str(e)}"
 
-    # Return the first (and only) line from the output
     return output_lines[0] if output_lines else "No output produced."
 
 # ---------- Gradio Interface ----------
@@ -131,14 +127,14 @@ description = (
 iface = gr.Interface(
     fn=diacritize_text,
     inputs=[
-        gr.inputs.Dropdown(
+        gr.Dropdown(
             choices=["Encoder-Decoder", "Encoder-Only"],
-            default="Encoder-Only",
+            value="Encoder-Only",
             label="Model Selection"
         ),
-        gr.inputs.Textbox(lines=4, placeholder="Enter Arabic text here...", label="Input Text")
+        gr.Textbox(lines=4, placeholder="Enter Arabic text here...", label="Input Text")
     ],
-    outputs=gr.outputs.Textbox(label="Diacritized Output"),
+    outputs=gr.Textbox(label="Diacritized Output"),
     title=title,
     description=description,
     allow_flagging="never"