Update app.py

app.py

@@ -3,8 +3,8 @@ from transformers import RobertaTokenizer, AutoModelForSequenceClassification
 import torch
 
 # Chargement des modèles
-state_dict_comment = torch.load("fine_tuned_roberta_comment.bin", map_location=torch.device("cpu"))
-state_dict_full = torch.load("fine_tuned_roberta_full.bin", map_location=torch.device("cpu"))
+state_dict_comment = torch.load("fine_tuned_roberta_comments.bin", map_location=torch.device("cpu"))
+state_dict_full = torch.load("fine_tuned_roberta.bin", map_location=torch.device("cpu"))
 
 tokenizer = RobertaTokenizer.from_pretrained("roberta-base")
 model_comment = AutoModelForSequenceClassification.from_pretrained('roberta-base', problem_type="multi_label_classification", num_labels=3)