Add app file

app.py

@@ -0,0 +1,70 @@
+import gradio as gr
+from huggingface_hub import hf_hub_download
+import importlib.util
+import sys
+
+# =========================================
+# 1. Load the Inference Script Dynamically
+# =========================================
+REPO_ID = "malkhuzanie/arabic-punctuation-checkpoints"
+FILENAME = "inference.py"
+
+print(f"Downloading {FILENAME} from {REPO_ID}...")
+script_path = hf_hub_download(repo_id=REPO_ID, filename=FILENAME)
+
+# Load the script as a module
+spec = importlib.util.spec_from_file_location("inference", script_path)
+inference = importlib.util.module_from_spec(spec)
+
+# CRITICAL: Register the module so pickle can find 'Vocabulary' class
+sys.modules["inference"] = inference 
+
+spec.loader.exec_module(inference)
+print("Inference script loaded successfully!")
+
+# =========================================
+# 2. Initialize Model
+# =========================================
+# The class inside inference.py handles downloading the weights/vocab automatically
+restorer = inference.PunctuationRestorer(repo_id=REPO_ID)
+
+# =========================================
+# 3. Define Gradio Interface
+# =========================================
+def gradio_predict(text):
+    if not text or not text.strip():
+        return ""
+    return restorer.predict(text)
+
+# Custom CSS to force Right-to-Left (RTL) alignment for Arabic
+css = """
+textarea { direction: rtl; }
+#component-0 { direction: rtl; }
+#component-1 { direction: rtl; }
+"""
+
+iface = gr.Interface(
+    fn=gradio_predict,
+    inputs=gr.Textbox(
+        lines=3, 
+        placeholder="اكتب النص العربي هنا...", 
+        label="Input Text (النص المدخل)",
+        text_align="right",
+        rtl=True
+    ),
+    outputs=gr.Textbox(
+        label="Punctuated Text (النص المرقّم)",
+        text_align="right",
+        rtl=True,
+        show_copy_button=True
+    ),
+    title="Arabic Punctuation Restoration",
+    description="Enter raw Arabic text without punctuation to see the model predictions.",
+    examples=[
+        ["هل تساءلت يوما عن معنى الحياة ما هي الأسئلة التي تشغل بالك"],
+        ["الجو جميل اليوم لا اعتقد انها ستمطر"]
+    ],
+    theme="default"
+)
+
+iface.launch()

requirements.txt

@@ -0,0 +1,3 @@
+torch
+huggingface-hub
+numpy