Update app.py

app.py

@@ -15,32 +15,37 @@ def query(payload):
 
 
 def analyze_sentiment(text):
-    output = query({     
+    output = query({
     "inputs":   f'''<|begin_of_text|>
     <|start_header_id|>system<|end_header_id|>
-you are a feeling analyser and you'll say only "positive" if i'm feeling positive and "negativ" if i'm feeling sad or bad   <|eot_id|>
+you are a feeling analyser and you'll say only "positive" if i'm feeling positive and "negativ" if i'm feeling sad   <|eot_id|>
     <|start_header_id|>user<|end_header_id|>
     {text}
     <|eot_id|>
     <|start_header_id|>assistant<|end_header_id|>
-
-    "parameters": {
-        "max_new_tokens": 1,
-        "return_full_text": False
-        }
     '''
 })
 
     # 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')
+    if isinstance(output, list) and len(output) > 0:
+        response = output[0].get('generated_text', '').strip().lower()
+    if 'positive' in response:
+       return 'positive'
+    elif 'negative' in response:
+       return 'negative'
     else:
-        return "Erreur: Réponse inattendue de l'API"
+       return "Erreur: Réponse inattendue"
+
+
+
+
 
 demo = gr.Interface(
-    fn = analyze_sentiment,
+    fn = query,
     inputs=["text"],
-    outputs=["text"],
+    outputs=["text"]
 )
 
 demo.launch()