Update app.py

app.py

@@ -1,28 +1,30 @@
+
 import requests
 import gradio as gr
-import os
+import spaces 
+import os 
+
 
 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)
-    if response.status_code == 200:
-        return response.json()
-    else:
-        raise Exception(f"Request failed with status code {response.status_code}: {response.text}")
+    return response.json()
 
 def detect_sentiment(message):
     prompt = f"Détecte le sentiment de ce message. Réponds par 'positif' ou 'négatif' :\nMessage : \"{message}\""
     response = query({"inputs": prompt})
-    generated_texts = response.get('generated_text', [])
-    if generated_texts:
-        sentiment = generated_texts[0].strip().lower()
-        return sentiment
-    else:
-        raise Exception("No generated_text found in API response")
+    sentiment = response[0]['generated_text'].strip().lower()
+    return sentiment
 
 def sentiment_analysis_interface(message):
     sentiment = detect_sentiment(message)
@@ -38,4 +40,6 @@ iface = gr.Interface(
 )
 
 # Lancer l'interface
-iface.launch()
+
+
+iface.launch()