Update app.py

app.py

@@ -33,10 +33,19 @@ def predict(text, chosen_model):
     # Initialize the pipeline with the chosen model
     model_pipeline = pipeline("text-classification", model=chosen_model)
     result = model_pipeline(text)
-    label = result[0]['label']
-    score = result[0]['score']
     
-    return label#, score
+    predicted_label = result[0]['label']
+    predicted_score = result[0]['score']
+
+    non_predicted_label = "Hate Speech" if predicted_label == "Non Hate Speech" else "Non Hate Speech"
+    non_predicted_score = 1 - predicted_score
+
+    scores_dict = {
+        predicted_label: predicted_score,
+        non_predicted_label: non_predicted_score
+    }
+    
+    return scores_dict#, predicted_label
 
 inputs = [
     gr.Textbox(label="Text", value= app_examples[0][0]),
@@ -45,7 +54,6 @@ inputs = [
 
 outputs = [
     gr.Label(label="Result"),
-    #gr.Markdown()
 ]