app.py

# Utiliser un pipeline comme helper de haut niveau
from transformers import pipeline
from huggingface_hub import login
import gradio as gr


# Initialiser le pipeline de génération de texte
pipe = pipeline("text-generation", model="mistralai/Mixtral-8x7B-Instruct-v0.1")

# Fonction pour générer une réponse
def generate_response(message):
    response = pipe(message)
    return response[0]['generated_text']

# Créer l'interface Gradio
interface = gr.Interface(
    fn=generate_response,
    inputs="text",
    outputs="text",
    title="Générateur de Texte Mixtral",
    description="Entrez un message et obtenez une réponse générée par le modèle Mixtral."
)

# Lancer l'interface
interface.launch()