Update app.py

app.py

@@ -1,8 +1,8 @@
-import torch
 import os 
 import requests
 import spaces 
 import gradio as gr 
+import re 
 
 api_token = os.environ.get("TOKEN")
 
@@ -16,7 +16,6 @@ def query(payload):
 	return response.json()
 	
 
-
 def analyze_sentiment(text):
     prompt = f"Tu es un analyseur de sentiment. Ton rôle est d'évaluer le sentiment général du texte fourni. Réponds uniquement par 'positif' ou 'négatif'. N'ajoute aucune explication. Voici le texte à analyser : {text}"
     
@@ -24,15 +23,17 @@ def analyze_sentiment(text):
         "inputs": prompt,
     })
     
-    # Assurez-vous de gérer correctement la sortie de l'API
     if isinstance(output, list) and len(output) > 0:
-        return output[0].get('generated_text', 'Erreur: Réponse inattendue')
+        response = output[0].get('generated_text', '')
+        # Chercher 'positif' ou 'négatif' dans la réponse
+        sentiment = re.search(r'\b(positif|négatif)\b', response.lower())
+        if sentiment:
+            return sentiment.group()
+        else:
+            return "Sentiment non détecté"
     else:
         return "Erreur: Réponse inattendue de l'API"
 
-
-
-
 demo = gr.Interface(
     fn = analyze_sentiment,
     inputs=["text"],