import gradio as gr
from transformers import pipeline
# Définir le modèle à utiliser
model_name = "salmapm/llama2_salma"

# Sélectionner le pipeline (l'authentification est gérée par le CLI Hugging Face)
generator = pipeline("text-generation", model=model_name)

# Fonction de génération de texte
def generate_text(prompt):
    results = generator(prompt, max_length=50)
    return results[0]['generated_text']

# Créer l'interface utilisateur avec Gradio
iface = gr.Interface(
    fn=generate_text,
    inputs="text",
    outputs="text",
    title="Générateur de Texte",
    description="Entrez un texte et laissez le modèle générer la suite."
)

# Lancer l'application
iface.launch()