Update app.py
Browse files
app.py
CHANGED
@@ -27,6 +27,9 @@ def load_model():
|
|
27 |
return trainer, model, tokenizer
|
28 |
|
29 |
def align_word_ids(texts):
|
|
|
|
|
|
|
30 |
# Utilisez le tokenizer pour obtenir les tokens de chaque mot
|
31 |
tokenized_inputs = tokenizer(texts, padding='max_length', max_length=218, truncation=True, return_tensors="pt")
|
32 |
input_ids = tokenized_inputs["input_ids"][0]
|
|
|
27 |
return trainer, model, tokenizer
|
28 |
|
29 |
def align_word_ids(texts):
|
30 |
+
|
31 |
+
trainer, model, tokenizer = load_model()
|
32 |
+
|
33 |
# Utilisez le tokenizer pour obtenir les tokens de chaque mot
|
34 |
tokenized_inputs = tokenizer(texts, padding='max_length', max_length=218, truncation=True, return_tensors="pt")
|
35 |
input_ids = tokenized_inputs["input_ids"][0]
|