Spaces:
Sleeping
Sleeping
File size: 721 Bytes
8ff930b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 |
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."
|