TunisAI-Agent / app.py
ghariani's picture
Update app.py
d6bf15e verified
raw
history blame contribute delete
824 Bytes
import gradio as gr
from transformers import pipeline
from langdetect import detect
model_name = "aubmindlab/aragpt2-base"
chatbot = pipeline("text-generation", model=model_name)
def reply(prompt):
if prompt.strip() == "":
return "يرجى إدخال نص."
try:
lang = detect(prompt)
except:
lang = "ar"
response = chatbot(prompt, max_length=100, do_sample=True)[0]["generated_text"]
return response.strip()
interface = gr.Interface(
fn=reply,
inputs=gr.Textbox(label="📝 أدخل نصًا بأي لغة"),
outputs=gr.Textbox(label="💬 رد الوكيل"),
title="🌍 TunisAI-Agent — ذكاء صناعي متعدد اللغات",
description="وكيل ذكاء صناعي مجاني يعمل بأي لغة 🧠🌐"
)
interface.launch(share=True)