Hugging Face's logo

Spaces:
dinalzein
/
LanguageIdentifier
Runtime error

App Files Community
dinalzein commited on
Commit
fafbc3d
·
1 Parent(s): 30e66c5

update model name

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import torch
4
  import numpy as np
5
  from mapping_labels import languages_map, id2label
6
 
7
- model_checkpoint = "dinalzein/xlm-roberta-base-finetuned-language-detection-new"
8
  tokenizer = AutoTokenizer.from_pretrained(model_checkpoint, use_fast=True)
9
  model = AutoModelForSequenceClassification.from_pretrained(model_checkpoint)
10
  trainer = Trainer(model)
 
4
  import numpy as np
5
  from mapping_labels import languages_map, id2label
6
 
7
+ model_checkpoint = "dinalzein/xlm-roberta-base-finetuned-language-identification"
8
  tokenizer = AutoTokenizer.from_pretrained(model_checkpoint, use_fast=True)
9
  model = AutoModelForSequenceClassification.from_pretrained(model_checkpoint)
10
  trainer = Trainer(model)