app.py

# Use a pipeline as a high-level helper
from transformers import pipeline
import gradio as gr


pipe = pipeline("text2text-generation", model="basharalrfooh/Fine-Tashkeel")

def diacratize_text(text):
    generated_text = pipe(text)
    return' '.join([s['generated_text'] for s in generated_text])

demo = gr.Interface(
    fn=diacratize_text,
    inputs=["text"],
    outputs=["text"],
)

demo.launch()