Spaces:

Abelex
/

sentence

Runtime error

Abelex commited on 28 days ago

Commit

2fd76cc

verified ·

1 Parent(s): 543f953

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,7 @@ DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 # MODEL CONFIG (MUST MATCH TRAINING)
 # -------------------------------------------------
 PRETRAINED = "Davlan/bert-base-multilingual-cased-finetuned-amharic"
-HF_MODEL_ID = "Abelex/SentenceChunking-AfriBerta-Afanoromo"
 CHUNK_SIZE = 512
 MAX_CHUNKS = 8

 # MODEL CONFIG (MUST MATCH TRAINING)
 # -------------------------------------------------
 PRETRAINED = "Davlan/bert-base-multilingual-cased-finetuned-amharic"
+HF_MODEL_ID = "Abelex/afro-xlmr-large"
 CHUNK_SIZE = 512
 MAX_CHUNKS = 8