Update app.py

app.py

@@ -1,38 +1,33 @@
+import torch
 import os 
 import requests
 import spaces 
 import gradio as gr 
-import re 
 
 api_token = os.environ.get("TOKEN")
 
-
 API_URL = "https://api-inference.huggingface.co/models/meta-llama/Meta-Llama-3-8B-Instruct"
 headers = {"Authorization": f"Bearer {api_token}"}
+
 @spaces.GPU
 
 def query(payload):
-	response = requests.post(API_URL, headers=headers, json=payload)
-	return response.json()
-	
+    response = requests.post(API_URL, headers=headers, json=payload)
+    return response.json()
+
 def analyze_sentiment(text):
-    prompt = f"Analyse le sentiment de ce texte et réponds uniquement par 'positif' ou 'négatif' sans aucune autre explication : {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}"
+
     output = query({
         "inputs": prompt,
     })
-    
+
+    # Assurez-vous de gérer correctement la sortie de l'API
     if isinstance(output, list) and len(output) > 0:
-        response = output[0].get('generated_text', '').strip().lower()
-        if 'positif' in response:
-            return "positif"
-        elif 'négatif' in response:
-            return "négatif"
-        else:
-            return "Sentiment non détecté"
+        return output[0].get('generated_text', 'Erreur: Réponse inattendue')
     else:
         return "Erreur: Réponse inattendue de l'API"
-    
+
 demo = gr.Interface(
     fn = analyze_sentiment,
     inputs=["text"],