Update core/extractor.py

core/extractor.py

@@ -9,13 +9,13 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 class Entity(BaseModel):
     id: str = Field(description="ID unique (ex: E1).")
     name: str = Field(description="Nom de l'entité.")
-    type: str = Field(description="Type d'entité (ex: Person, Metric, Model, Zone, Concept).")
+    type: str = Field(description="Type d'entité.")
     description: str = Field(description="Contexte ou rôle.")
 
 class Relationship(BaseModel):
     source: str = Field(alias="from", description="ID de l'entité source.")
     target: str = Field(alias="to", description="ID de l'entité cible.")
-    type: str = Field(description="Action ou lien sémantique (ex: USES, CORRELATED_WITH, AUTHORED_BY).")
+    type: str = Field(description="Action ou lien sémantique.")
     description: str = Field(description="Explication du lien ou Détails.")
 
 class KnowledgeGraph(BaseModel):
@@ -26,7 +26,7 @@ class ExtractorEngine:
     def __init__(self):
         self.model_name = "Qwen/Qwen2.5-1.5B-Instruct"
         if 'llm_model' not in st.session_state:
-            with st.spinner("🚀 Chargement initial du cerveau IA..."):
+            with st.spinner("Chargement initial du cerveau IA..."):
                 st.session_state.llm_tokenizer = AutoTokenizer.from_pretrained(self.model_name)
                 st.session_state.llm_model = AutoModelForCausalLM.from_pretrained(
                     self.model_name, torch_dtype=torch.float32, device_map="cpu", low_cpu_mem_usage=True