Views
No views yet
gerulata/slovakbert, which aligns well with our model’s requirements.1from transformers import RobertaTokenizer, AutoModelForSeq2SeqLM
2
3# Load pretrained tokenizer and model
4tokenizer = RobertaTokenizer.from_pretrained(
5 'gerulata/slovakbert',
6 weights_only=False,
7 token="###YOUR_HF_TOKEN###"
8)
9model = AutoModelForSeq2SeqLM.from_pretrained(
10 "timotejKralik/hate_speech_correction_slovak",
11 weights_only=False,
12 token="###YOUR_HF_TOKEN###"
13)
14
15# Input text containing potentially harmful language
16input_text = "Opač jak ši sebe obľik tote nohavky, ši jak mantak."
17print(f"Input: {input_text}")
18
19# Tokenize input and generate output
20inputs = tokenizer(input_text, return_tensors="pt")
21outputs = model.generate(**inputs)
22output_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
23
24print("Output:", output_text)