app.py

import gradio as gr
import random
from transformers import pipeline

fill_mask = pipeline(
    "fill-mask",
    model="alger-ia/dziribert",
    tokenizer="alger-ia/dziribert"
)

def greet(prompt="أنا", maxTokens=50):
    generated_text_set = set()
    text = prompt +" [MASK]"
    for i in range(int(maxTokens)):
        result = fill_mask(text)
        r = random.choice(result)["sequence"]
        if r not in generated_text_set:
            print(r)
            text = r + " [MASK]"
            generated_text_set.add(r)
        else:
            continue
    return text

iface = gr.Interface(fn=greet, inputs=["text", "number"], outputs="text", examples=[["أنا [MASK]",50]] )
iface.launch()