Update app.py

app.py

@@ -1,44 +1,59 @@
-import gradio as gr
-import requests
 import os
-import spaces 
-
+import requests
+import gradio as gr
 
-API_URL = "https://api-inference.huggingface.co/models/meta-llama/Meta-Llama-3-8B-Instruct"
 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()
 
-def generate_response(prompt):
-    payload = {
-        "inputs": prompt,
-        "parameters": {
-            "max_new_tokens": 100,
-            "temperature": 0.7,
-            "top_p": 0.95,
-            "do_sample": True
-        }
-    }
-    
-    response = query(payload)
-    
-    if isinstance(response, list) and len(response) > 0:
-        return response[0].get('generated_text', '')
-    elif isinstance(response, dict) and 'generated_text' in response:
-        return response['generated_text']
-    return "Désolé, je n'ai pas pu générer de réponse."
-
-def chatbot(message, history):
-    response = generate_response(message)
-    return response
-
-iface = gr.ChatInterface(
-    fn=chatbot,
-    title="Chatbot Meta-Llama-3-8B-Instruct",
-    description="Interagissez avec le modèle Meta-Llama-3-8B-Instruct."
+def analyze_sentiment(text):
+    output = query({
+        "inputs": f'''<|begin_of_text|>
+<|start_header_id|>system<|end_header_id|>
+You're going to analyze the texts I'm going to give you and you're only going to tell me which category they belong to by answering only the words that correspond to the following categories: for posts that talk about chat models/LLM you'll return "Chatmodel/LLM", for posts that talk about image generation models you'll return "image_generation", for texts that ask for information from the community you'll return "questions".For texts about recent discoveries that don't fit into the previous categories, you'll return "other".
+
+<|eot_id|>
+<|start_header_id|>user<|end_header_id|>
+{text}
+<|eot_id|>
+<|start_header_id|>assistant<|end_header_id|>
+
+'''
+    })
+
+
+
+    if isinstance(output, list) and len(output) > 0:
+        response = output[0].get('generated_text', '').strip().lower()
+
+
+        questions = response.count('questions')
+        ChatmodelLLM = response.count('Chatmodel/LLM')
+        other = response.count('other')
+        image_generation = response.count("image_generation")
+
+    if questions == 2:
+        return 'questions'
+    elif ChatmodelLLM == 2:
+        return 'Chat Model/LLM'
+    elif other == 2 : 
+        return "Other"
+    elif image_generation == 2 : 
+        return "Image Generation"
+    else :
+        return f"Erreur: Réponse ambiguë - '{response}'"
+
+
+demo = gr.Interface(
+    fn=analyze_sentiment,
+    inputs="text",
+    outputs="text"
 )
 
-iface.launch()
+demo.launch()