Update app.py

app.py

@@ -1,17 +1,25 @@
 # Utiliser un pipeline comme helper de haut niveau
 from transformers import pipeline
 from huggingface_hub import login
+import gradio as gr
 
-# Se connecter à Hugging Face avec votre token
-
-# Définir le message comme une chaîne de caractères
-message = "Who are you?"
 
 # Initialiser le pipeline de génération de texte
 pipe = pipeline("text-generation", model="mistralai/Mixtral-8x7B-Instruct-v0.1")
 
-# Générer une réponse
-response = pipe(message)
+# 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."
+)
 
-# Afficher la réponse
-print(response)
+# Lancer l'interface
+interface.launch()