Update app.py

app.py

@@ -2,23 +2,23 @@ import gradio as gr
 import requests
 import spaces 
 
-API_URL = "https://api-inference.huggingface.co/models/meta-llama/Meta-Llama-3-8B-Instruct"
+API_URL = "https://api-inference.huggingface.co/models/distilbert-base-uncased-finetuned-sst-2-english"
 headers = {"Authorization": "Bearer hf_xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"}
 
 @spaces.GPU
 
 def analyze_sentiment(text):
-    payload = {
-        "inputs": f"Analyze the sentiment of the following text and respond with either 'heureux' or 'malheureux': {text}"
-    }
+    payload = {"inputs": text}
     try:
         response = requests.post(API_URL, headers=headers, json=payload)
-        response.raise_for_status()  # Vérifie si la requête a réussi
+        response.raise_for_status()
         result = response.json()
         
-        if isinstance(result, list) and len(result) > 0 and 'generated_text' in result[0]:
-            sentiment = result[0]['generated_text'].strip().lower()
-            return "heureux" if "heureux" in sentiment else "malheureux"
+        if isinstance(result, list) and len(result) > 0 and isinstance(result[0], list):
+            # Le modèle renvoie généralement une liste de scores pour les étiquettes NEGATIVE et POSITIVE
+            sentiment_scores = result[0]
+            sentiment = "heureux" if sentiment_scores[1] > sentiment_scores[0] else "malheureux"
+            return sentiment
         else:
             return "Erreur: Format de réponse inattendu"
     except requests.exceptions.RequestException as e: