Spaces:
Sleeping
Sleeping
import spaces | |
from transformers import pipeline | |
# Chargement du modèle de classification de sentiment | |
classifier = pipeline("sentiment-analysis") | |
# Fonction pour prédire le sentiment d'un post | |
def predict_sentiment(post): | |
result = classifier(post) | |
sentiment = result[0]['label'] | |
return sentiment | |
# Fonction de génération de texte basée sur le sentiment | |
def generate_response(prompt): | |
sentiment = predict_sentiment(prompt) | |
if sentiment == 'POSITIVE': | |
return "Je suis content de discuter avec vous!" | |
elif sentiment == 'NEGATIVE': | |
return "Je suis désolé que vous ressentiez cela. Comment puis-je vous aider?" | |
else: | |
return "Merci pour vos commentaires." | |