alex-abb commited on
Commit
8ff930b
1 Parent(s): 8b7e59f
Files changed (1) hide show
  1. app.py +30 -0
app.py ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import spaces
2
+ from transformers import pipeline
3
+
4
+ # Chargement du modèle de classification de sentiment
5
+
6
+ classifier = pipeline("sentiment-analysis")
7
+
8
+ # Fonction pour prédire le sentiment d'un post
9
+
10
+ def predict_sentiment(post):
11
+ result = classifier(post)
12
+ sentiment = result[0]['label']
13
+ return sentiment
14
+
15
+
16
+ # Fonction de génération de texte basée sur le sentiment
17
+
18
+ def generate_response(prompt):
19
+ sentiment = predict_sentiment(prompt)
20
+ if sentiment == 'POSITIVE':
21
+ return "Je suis content de discuter avec vous!"
22
+ elif sentiment == 'NEGATIVE':
23
+ return "Je suis désolé que vous ressentiez cela. Comment puis-je vous aider?"
24
+ else:
25
+ return "Merci pour vos commentaires."
26
+
27
+
28
+
29
+
30
+