alex-abb commited on
Commit
e1021d2
1 Parent(s): cf4674b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -9
app.py CHANGED
@@ -3,22 +3,25 @@ from transformers import pipeline
3
  import spaces
4
 
5
  # Charger le pipeline de classification des sentiments
6
- sentiment_analyzer = pipeline("sentiment-analysis", model="distilbert-base-uncased-finetuned-sst-2-english")
7
 
8
- @spaces.GPU
 
 
 
9
 
10
  def analyze_sentiment(text):
11
  # Analyser le sentiment du texte
12
- result = sentiment_analyzer(text)
13
- # Retourner le résultat sous forme de texte
14
- sentiment = result[0]['label']
15
- score = result[0]['score']
16
- return f"Sentiment: {sentiment}, Score: {score:.2f}"
 
17
 
18
  # Créer l'interface Gradio
19
  interface = gr.Interface(
20
- fn=analyze_sentiment,
21
- inputs="text",
22
  outputs="text",
23
  title="Analyse des Sentiments",
24
  description="Entrez un texte pour analyser son sentiment (positif, négatif ou neutre)."
 
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")@spaces.GPU
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)."