alex-abb commited on
Commit
efba1b1
1 Parent(s): a998401

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -10
app.py CHANGED
@@ -2,29 +2,33 @@ import gradio as gr
2
  from transformers import pipeline
3
  import spaces
4
 
5
- # Charger le pipeline de classification des sentiments
6
-
7
-
8
- pipe = pipeline("text-generation", model="deepseek-ai/deepseek-llm-7b-base")
9
-
10
  @spaces.GPU
11
 
 
 
 
12
  def analyze_sentiment(text):
13
  # Analyser le sentiment du texte
14
- result = pipe(text)
 
 
15
  for result in results:
16
  label = result['label']
17
  score = result['score']
18
  sentiments.append(f"Sentiment: {label}, Score: {score:.2f}")
19
  return "\n".join(sentiments)
20
 
 
 
 
 
21
  # Créer l'interface Gradio
22
  interface = gr.Interface(
23
- fn=pipe,
24
- inputs=gr.inputs.Textbox(lines=5, placeholder="Entrez votre texte ici..."),
25
- outputs="text",
26
  title="Analyse des Sentiments",
27
- description="Entrez un texte pour analyser son sentiment (positif, négatif ou neutre)."
28
  )
29
 
30
  # Lancer l'interface dans un Space Hugging Face
 
2
  from transformers import pipeline
3
  import spaces
4
 
 
 
 
 
 
5
  @spaces.GPU
6
 
7
+ # Charger le pipeline d'analyse des sentiments
8
+ sentiment_analyzer = pipeline("text-classification", model="mrm8488/deberta-v3-base-finetuned-mnli")
9
+
10
  def analyze_sentiment(text):
11
  # Analyser le sentiment du texte
12
+ results = sentiment_analyzer(text)
13
+ # Formater les résultats pour l'affichage
14
+ sentiments = []
15
  for result in results:
16
  label = result['label']
17
  score = result['score']
18
  sentiments.append(f"Sentiment: {label}, Score: {score:.2f}")
19
  return "\n".join(sentiments)
20
 
21
+ # Définir les composants d'entrée et de sortie Gradio
22
+ inputs = gr.Textbox(lines=5, label="Entrez votre texte ici")
23
+ outputs = gr.Textbox(label="Résultat de l'analyse des sentiments")
24
+
25
  # Créer l'interface Gradio
26
  interface = gr.Interface(
27
+ fn=analyze_sentiment,
28
+ inputs=inputs,
29
+ outputs=outputs,
30
  title="Analyse des Sentiments",
31
+ description="Entrez un texte pour analyser son sentiment (positif, négatif, ou neutre). Le modèle utilisé est un modèle généraliste finement ajusté pour la classification des sentiments."
32
  )
33
 
34
  # Lancer l'interface dans un Space Hugging Face