code

app.py

@@ -11,11 +11,12 @@ cpv = pd.read_csv("cpv.csv", sep=";")
 df = pd.read_csv("code-desc.csv", sep=";")
 
 labels = df.columns[1:]
-cpv = cpv.columns[1:]
+cpv = cpv.code
 
 id2label = {idx:label for idx, label in enumerate(labels)}
 label2id = {label:idx for idx, label in enumerate(labels)}
-    
+codeDesc = dict(zip(df.code, df.desc))
+   
 def askcpv(description):
     
     encoding = tokenizer(description, return_tensors="pt")