Mohssinibra commited on
Commit
8dc5e68
·
verified ·
1 Parent(s): bc32c3a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -35,7 +35,7 @@ bert_model = BertForSequenceClassification.from_pretrained(bert_model_name, num_
35
  # Charger le modèle et le tokenizer Darija
36
  sentiment_model_name = "BenhamdaneNawfal/sentiment-analysis-darija"
37
  sentiment_tokenizer = AutoTokenizer.from_pretrained(sentiment_model_name)
38
- sentiment_model = AutoModelForSequenceClassification.from_pretrained(sentiment_model_name,num_labels=3).to("cuda" if torch.cuda.is_available() else "cpu")
39
 
40
  # Labels du modèle (à modifier selon le modèle utilisé)
41
  sentiment_labels = ["Négatif", "Neutre", "Positif"]
 
35
  # Charger le modèle et le tokenizer Darija
36
  sentiment_model_name = "BenhamdaneNawfal/sentiment-analysis-darija"
37
  sentiment_tokenizer = AutoTokenizer.from_pretrained(sentiment_model_name)
38
+ sentiment_model = AutoModelForSequenceClassification.from_pretrained(sentiment_model_name,num_labels=3,ignore_mismatched_sizes=True).to("cuda" if torch.cuda.is_available() else "cpu")
39
 
40
  # Labels du modèle (à modifier selon le modèle utilisé)
41
  sentiment_labels = ["Négatif", "Neutre", "Positif"]