kambris commited on
Commit
254cb4f
·
verified ·
1 Parent(s): ebe462c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,8 +8,8 @@ import torch
8
  tokenizer = T5Tokenizer.from_pretrained("UBC-NLP/araT5-base")
9
  model = T5ForConditionalGeneration.from_pretrained("UBC-NLP/araT5-base")
10
 
11
- # Emotion classification pipeline for Arabic (use an Arabic emotion classification model)
12
- emotion_classifier = pipeline("text-classification", model="d0r13n/ara-bert-base-arabic-emotion")
13
 
14
  # Function to get embeddings from ARAT5 for topic modeling
15
  def generate_embeddings(texts):
 
8
  tokenizer = T5Tokenizer.from_pretrained("UBC-NLP/araT5-base")
9
  model = T5ForConditionalGeneration.from_pretrained("UBC-NLP/araT5-base")
10
 
11
+ # Emotion classification pipeline (updated model for Arabic emotion classification)
12
+ emotion_classifier = pipeline("text-classification", model="aubmindlab/bert-base-arabertv2")
13
 
14
  # Function to get embeddings from ARAT5 for topic modeling
15
  def generate_embeddings(texts):