Update app.py

app.py

@@ -1,9 +1,33 @@
 import gradio as gr
 import os
+import requests
 
+# Retrieve the Hugging Face API token from the environment variable
 hf_token = os.getenv("HF_TOKEN")
 
-if hf_token is None:
-    print("No token provided")
+api_url = "https://api-inference.huggingface.co/models/SARAL-Influencer-OS/comment_analyser"
+
+headers = {"Authorization": f"Bearer {hf_token}"} if hf_token else {}
+
+label_map = {
+    "LABEL_0": "Buying Intent",
+    "LABEL_1": "Neutral Intent",
+    "LABEL_2": "Negative Intent"
+}
+
+def predict_comment(comment):
+    response = requests.post(api_url, headers=headers, json={"inputs": comment})
+    output = response.json()
     
-gr.load("models/SARAL-Influencer-OS/comment_analyser",hf_token=hf_token).launch()
+    # Map the model output to the corresponding label names
+    predictions = {label_map[item['label']]: item['score'] for item in output[0]}
+    return predictions
+
+iface = gr.Interface(
+    fn=predict_comment,
+    inputs=gr.inputs.Textbox(lines=2, placeholder="Enter your comment..."),
+    outputs=gr.outputs.Label(num_top_classes=3)
+)
+
+# Launch the interface
+iface.launch()