Update app.py

app.py

@@ -72,6 +72,11 @@ from transformers import AutoTokenizer
 tokenizer_id = "xlm-roberta-base"
 tokenizer = AutoTokenizer.from_pretrained(tokenizer_id)
 
+# get labels
+id2label = model.config.id2label
+label2id = model.config.label2id
+num_labels = len(id2label)
+
 # APP outputs 
 def app_outputs(uploaded_pdf):
     filename, msg, images = pdf_to_images(uploaded_pdf)