alex-abb's picture
Update app.py
d97ee44 verified
raw
history blame
456 Bytes
from transformers import pipeline
import gradio as gr
# Charger le modèle pour la génération de texte
pipe = pipeline("text-generation", model="MTSAIR/MultiVerse_70B")
# Fonction pour générer une réponse à partir du message de l'utilisateur
def generate_response(message):
response = pipe(message)
return response[0]['generated_text']
# Configurer et lancer l'interface de chat avec Gradio
gr.ChatInterface(fn=generate_response).launch()