import gradio as gr
from transformers import pipeline

# Charger le modèle directement depuis Hugging Face
model_name = "salmapm/llama2_salma"
generator = pipeline("text-generation", model=model_name)

# Définir une 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()