What would the script be like if the script for the original model was like this?

#16
by mohamed101020 - opened

from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
import pandas as pd

if name == "main":

text = "كيف الحال"

model_name = "basharalrfooh/Fine-Tashkeel"

tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForSeq2SeqLM.from_pretrained(model_name)

input_ids = tokenizer(text, return_tensors="pt").input_ids
outputs = model.generate(input_ids, max_new_tokens=128)
decoded_output = tokenizer.decode(outputs[0], skip_special_tokens=True, clean_up_tokenization_spaces=False)
print("Generated output:", decoded_output)

Your need to confirm your account before you can post a new comment.

Sign up or log in to comment