import gradio as gr
from transformers import AutoModel, AutoTokenizer

# بارگذاری مدل و توکنایزر
model_name = "universitytehran/PersianMind-v1.0"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModel.from_pretrained(model_name)

# تعریف تابع برای پردازش ورودی
def process_input(text):
    inputs = tokenizer(text, return_tensors="pt")
    outputs = model(**inputs)
    return outputs.last_hidden_state.mean(dim=1).detach().numpy()

# ایجاد رابط کاربری با Gradio
iface = gr.Interface(
    fn=process_input,
    inputs="text",
    outputs="text",
    title="PersianMind-v1.0",
    description="این یک مدل برای پردازش متن فارسی است."
)

# اجرای رابط کاربری
iface.launch()