Hugging Face's logo

Spaces:
Sehamsaa
/
arabic-complaint-classifier
Sleeping

App Files Community
Sehamsaa commited on
Commit
e8fc8aa
·
verified ·
1 Parent(s): fc1f035

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -21,7 +21,7 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
21
  # ============================================================================
22
 
23
  MODEL_PATH = os.getenv("MODEL_PATH", "./saved_model")
24
- TOKENIZER_NAME = "UBC-NLP/MARBERT"
25
  MAX_LENGTH = 128 # نفس القيمة في التدريب الأصلي
26
 
27
  LABELS_EN = ["Low", "Medium", "High", "Critical"]
 
21
  # ============================================================================
22
 
23
  MODEL_PATH = os.getenv("MODEL_PATH", "./saved_model")
24
+ TOKENIZER_NAME = "aubmindlab/bert-base-arabertv02"
25
  MAX_LENGTH = 128 # نفس القيمة في التدريب الأصلي
26
 
27
  LABELS_EN = ["Low", "Medium", "High", "Critical"]