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()