Update app.py

app.py

@@ -5,33 +5,40 @@ import gradio as gr
 API_URL = "https://api-inference.huggingface.co/models/ProsusAI/finbert"
 headers = {"Authorization": "Bearer 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"}
 
-def query(payload):
-	response = requests.post(API_URL, headers=headers, json=payload)
-	return response.json()['label']
+# def query(payload):
+# 	response = requests.post(API_URL, headers=headers, json=payload)
+#     sentiment_prediction = response.json()
+
+#     pos, neg, neutr = sentiment_prediction[:, 0].item(), sentiment_prediction[:, 1].item(), sentiment_prediction[:, 2].item()
+#     sentiment_label = "Positive" if pos > neg and pos > neutr else "Negative" if neg > pos and neg > neutr else "Neutral"
+
+# 	return sentiment_label
 	
-output = query({
-	"inputs": "I like you. I love you",
-})
+# output = query({
+# 	"inputs": "I like you. I love you",
+# })
 
 
-# def predict_sentiment(payload):
-#     # Sentiment Analysis
+def predict_sentiment(payload):
+    # Sentiment Analysis
+    response = requests.post(API_URL, headers=headers, json=payload)
+    sentiment_prediction = response.json()
     
-#     with torch.no_grad():
-#         sentiment_outputs = requests.post(API_URL, headers=headers, json=payload)
-#         sentiment_prediction = torch.nn.functional.softmax(sentiment_outputs.logits, dim=-1)
+    # with torch.no_grad():
+    #     sentiment_outputs = requests.post(API_URL, headers=headers, json=payload)
+    #     sentiment_prediction = torch.nn.functional.softmax(sentiment_outputs.logits, dim=-1)
 
-#     pos, neg, neutr = sentiment_prediction[:, 0].item(), sentiment_prediction[:, 1].item(), sentiment_prediction[:, 2].item()
-#     sentiment_label = "Positive" if pos > neg and pos > neutr else "Negative" if neg > pos and neg > neutr else "Neutral"
+    pos, neg, neutr = sentiment_prediction[:, 0].item(), sentiment_prediction[:, 1].item(), sentiment_prediction[:, 2].item()
+    sentiment_label = "Positive" if pos > neg and pos > neutr else "Negative" if neg > pos and neg > neutr else "Neutral"
 
 
-#     return sentiment_label
+    return sentiment_label
 
 
 
 # Gradio Interface
 iface = gr.Interface(
-    fn=query,
+    fn=predict_sentiment,
     inputs=[gr.Textbox(lines=2, label="Financial Statement")],
     outputs=[
         gr.Textbox(label="Sentiment"),