التعرف على الكائنات

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 from models.space_a import summarize_question
 from models.space_b import generate_mentalqa_answer
-
+from models.space_er import extract_entities
 css = """
 body {
   direction: rtl;
@@ -47,10 +47,12 @@ with gr.Blocks(css=css) as demo:
             choices=[
                 ("تصنيف الأسئلة والأجوبة", "classification"),
                 ("تلخيص النصوص", "summarization"),
+                ("التعرف على الكيانات", "entity_recognition")
             ],
-            label="اختيار نوع المهمة",
-            value="classification"
-        )
+        label="اختيار نوع المهمة",
+        value="classification"
+)
+
 
         user_input = gr.Textbox(placeholder="أدخل النص هنا لتحليله...", lines=4, show_label=False)