alex-abb's picture
Update app.py
91149d9 verified
raw
history blame
505 Bytes
import transformers
from transformers import pipeline
import gradio as gr
import spaces
@spaces.GPU
# 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()