app.py

import gradio as gr
from transformers import pipeline

# Crée le générateur de texte avec un modèle francophone
generator = pipeline("text-generation", model="asi/gpt-fr-cased-small")

# Fonction qui génère du texte à partir du prompt utilisateur
def repondre(prompt):
    result = generator(
        prompt,
        max_length=60,
        num_return_sequences=1,
        temperature=0.9,
        top_k=50,
        repetition_penalty=1.2
    )
    return result[0]["generated_text"]

# Interface Gradio
interface = gr.Interface(
    fn=repondre,
    inputs="text",
    outputs="text",
    title="Chat avec GPT-2 (version française 🧠🇫🇷)",
    description="Pose une question ou commence une phrase, le modèle complète !"
)

# Lancement
interface.launch()