Create app.py

app.py

@@ -0,0 +1,78 @@
+import gradio as gr
+from sentence_transformers.cross_encoder import CrossEncoder
+import numpy as np
+
+# Load the model
+# Replace "ALJIACHI/Mizan-Rerank-v1" with your actual model name
+model = CrossEncoder("ALJIACHI/Mizan-Rerank-v1")
+
+def rerank_texts(query, texts):
+    """
+    Rerank a list of texts based on their relevance to the query.
+    
+    Args:
+        query (str): The search query
+        texts (str): Newline-separated texts to rank
+    
+    Returns:
+        str: Formatted results with scores
+    """
+    # Split the input texts by newline
+    text_list = [t.strip() for t in texts.split('\n') if t.strip()]
+    
+    # Create sentence pairs
+    sentence_pairs = [[query, text] for text in text_list]
+    
+    # Get scores from the model
+    scores = model.predict(sentence_pairs)
+    
+    # Create results with ranks, scores, and texts
+    results = [(score, text) for score, text in zip(scores, text_list)]
+    
+    # Sort by score in descending order
+    results.sort(reverse=True)
+    
+    # Format the output
+    output = ""
+    for i, (score, text) in enumerate(results, 1):
+        output += f"#{i} (Score: {score:.4f}): {text}\n\n"
+    
+    return output
+
+# Create Gradio interface
+demo = gr.Interface(
+    fn=rerank_texts,
+    inputs=[
+        gr.Textbox(label="Query", placeholder="Enter your search query here..."),
+        gr.Textbox(
+            label="Texts to Rank", 
+            placeholder="Enter texts to rank, one per line...",
+            lines=10
+        )
+    ],
+    outputs=gr.Textbox(label="Ranked Results"),
+    title="Arabic Text Reranking Demo",
+    description=(
+        "This demo uses the Mizan-Rerank-v1 model to rank texts based on their relevance to a query. "
+        "Enter your query and a list of texts (one per line), and the model will rank them by relevance."
+    ),
+    examples=[
+        [
+            "ما هو القانون الجديد بشأن الضرائب في 2024؟",
+            "نشرت الجريدة الرسمية قانوناً جديداً في 2024 ينص على زيادة الضرائب على الشركات الكبرى بنسبة 5%\nالضرائب تعد مصدراً مهماً للدخل القومي وتختلف نسبتها من دولة إلى أخرى.\nافتتحت الحكومة مشروعاً جديداً للطاقة المتجددة في 2024."
+        ],
+        [
+            "ما هو تفسير الآية وجعلنا من الماء كل شيء حي",
+            "تعني الآية أن الماء هو عنصر أساسي في حياة جميع الكائنات الحية، وهو ضروري لاستمرار الحياة.\nتم اكتشاف كواكب خارج المجموعة الشمسية تحتوي على مياه متجمدة.\nتحدث القرآن الكريم عن البرق والرعد في عدة مواضع مختلفة."
+        ],
+        [
+            "ما هي فوائد فيتامين د؟",
+            "يساعد فيتامين د في تعزيز صحة العظام وتقوية الجهاز المناعي، كما يلعب دوراً مهماً في امتصاص الكالسيوم.\nيستخدم فيتامين د في بعض الصناعات الغذائية كمادة حافظة.\nيمكن الحصول على فيتامين د من خلال التعرض لأشعة الشمس أو تناول مكملات غذائية."
+        ]
+    ],
+    allow_flagging="never"
+)
+
+# Launch the demo
+if __name__ == "__main__":
+    demo.launch()