Update app.py

app.py

@@ -12,16 +12,11 @@ model.eval()
 with open("child_to_parent_mapping.json", "r") as f:
     child_to_ancestor = json.load(f)
 
-with open(f"{model_path}/config.json", "r") as f:
-    config = json.load(f)
-    id2label = config["id2label"]
+id2label = model.config.id2label
 
-# Fonction d'extraction pour simuler une entrée formatée
 def extract_commit_text_hg_style(input_text):
-    # Ici, on pourrait simuler un vrai patch ou commit. Pour l’instant, on prend l’entrée brute.
     return input_text.strip()
 
-# Fonction Gradio de prédiction
 def predict_ancestors(input_text):
     text = extract_commit_text_hg_style(input_text)
     inputs = tokenizer(text, return_tensors="pt", truncation=True, padding="max_length", max_length=512)