Hugging Face's logo

Spaces:
Yxxi4
/
nahw-api
Sleeping

App Files Community
Yxxi4 commited on
Commit
a5492dc
·
verified ·
1 Parent(s): c2fc60c

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -141,7 +141,7 @@ tashkeel_model.eval()
141
 
142
  print("جاري تحميل نموذج AraBART...")
143
  # في حال كان مسار النموذج على Hugging Face مختلفاً، يرجى تعديل السطر التالي
144
- ARABART_MODEL_NAME = "UBC-NLP/AraBART-Morph-GEC"
145
  arabart_tokenizer = AutoTokenizer.from_pretrained(ARABART_MODEL_NAME)
146
  arabart_model = AutoModelForSeq2SeqLM.from_pretrained(ARABART_MODEL_NAME)
147
  gec_pipeline = pipeline("text2text-generation", model=arabart_model, tokenizer=arabart_tokenizer)
 
141
 
142
  print("جاري تحميل نموذج AraBART...")
143
  # في حال كان مسار النموذج على Hugging Face مختلفاً، يرجى تعديل السطر التالي
144
+ ARABART_MODEL_NAME = "CAMeL-Lab/arabart-qalb15-gec-ged-13"
145
  arabart_tokenizer = AutoTokenizer.from_pretrained(ARABART_MODEL_NAME)
146
  arabart_model = AutoModelForSeq2SeqLM.from_pretrained(ARABART_MODEL_NAME)
147
  gec_pipeline = pipeline("text2text-generation", model=arabart_model, tokenizer=arabart_tokenizer)