Add hate and non hate

app.py

@@ -351,9 +351,9 @@ def load_model(tweet):
             pred = model(inputs)
             print("prediction ",pred)
             print("sigmoid output",torch.sigmoid(pred))
-            print("sigmoid item",torch.sigmoid(pred).item())
+            pred = torch.sigmoid(pred).item()
             # Assuming your model returns a single value for prediction
-            pred = torch.round(torch.sigmoid(pred)).item()
+            
         
         return pred
 
@@ -367,20 +367,40 @@ def load_model(tweet):
 #Gradio interface
 def greet(tweet):
     print("start")
-    predicted_class = load_model(tweet)
+    prediction = load_model(tweet)
     # features_list = extract_features(tweet)
     # print(personality_detection(tweet))
     # print(str(features_list["Average_Word_Length"]))
     # print(calc_emotion_score(tweet))
+    predicted_class = torch.round(torch.sigmoid(prediction)).item()
     print("end")
+    if (predicted_class==0.0):
+        label = "Hate"
+    else:
+        label = "Non Hate"
     
 
-    return str(predicted_class)
+    return label,str(prediction)+"%",str(1-prediction)+"%"
 
 # demo = gr.Interface(fn=greet, inputs="text", outputs="text")
 demo = gr.Interface(
     title = "Unmasking Hate: An Integrated Approach to Detecting Hate Speech in Social Media",
     # fn=greet,
     fn=greet, inputs=gr.Textbox(placeholder="Enter an input sentence...",label="Input Sentence"), 
-    allow_flagging = "never",outputs="text")
+    allow_flagging = "never",outputs=[
+        gr.Label(label="Label"),
+        gr.Textbox(label="Hate Speech Percentage"),
+        gr.Textbox(label="Non Hate Speech Percentage")
+    ],
+    examples=[
+        ["I like you"],
+        ["I hate you"],
+        ["I can't stand those asian always causing trouble. They need to go back to where they came from!"],
+        ["Just saw a Sunni preaching on the street. Why don't they go worship in their own country instead of invading ours?"],
+        ["I wish all bisexuals would just disappear. Sick of their agenda being shoved in our faces"],
+        ["Had a great time celebrating diversity at the multicultural festival today!"],
+        ["Congratulations to Sri Lankans for their cultural contributions to our society"],
+        ["Love is love, no matter who you are or who you love"] ]
+        )
+
 demo.launch()