Update app.py

app.py

@@ -1,7 +1,6 @@
 import gradio as gr
 import torch
 import json
-import base64
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 
 model_path = "CIRCL/cwe-parent-vulnerability-classification-roberta-base"
@@ -36,13 +35,12 @@ def predict_ancestors(input_text):
         ancestor = child_to_ancestor.get(cwe_child, "N/A")
         results.append(f"{i}. CWE-{cwe_child} (ancestor: CWE-{ancestor}) - {score:.4f}")
 
-    return results
+    return "\n".join(results)
 
-# Interface Gradio
 gr.Interface(
     fn=predict_ancestors,
     inputs=gr.Textbox(label="Commit message or patch (e.g., 'hg')"),
-    outputs=gr.outputs.Textbox(label="Top 5 Predicted CWE Ancestors"),
+    outputs=gr.Textbox(label="Top 5 Predicted CWE Ancestors"),
     title="CWE Ancestor Predictor",
     description="Entrez un message de commit ou un patch. Le modèle prédit les 5 CWE enfants les plus probables et affiche leurs ancêtres."
 ).launch()