Views
No views yet
1from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
2import torch
3
4model = AutoModelForSeq2SeqLM.from_pretrained("Angelo25/Filipino-Lexical-Normalization")
5tokenizer = AutoTokenizer.from_pretrained("Angelo25/Filipino-Lexical-Normalization")
6model.eval()
7
8inputs = tokenizer("Sample Input Text", return_tensors="pt").to(model.device)
9output = model.generate(
10 **inputs,
11 max_new_tokens=inputs["input_ids"].shape[1] + 50,
12 num_beams=3,
13 early_stopping=True,
14 use_cache=True
15)
16print(tokenizer.decode(output[0], skip_special_tokens=True))