Add predicted_class

app.py

@@ -351,7 +351,7 @@ def load_model(tweet):
             pred = model(inputs)
             print("prediction ",pred)
             print("sigmoid output",torch.sigmoid(pred))
-            pred = torch.sigmoid(pred).item()
+            pred = torch.sigmoid(pred)
             # Assuming your model returns a single value for prediction
             
         
@@ -368,6 +368,7 @@ def load_model(tweet):
 def greet(tweet):
     print("start")
     prediction = load_model(tweet)
+    prediction_value = prediction.item()
     # features_list = extract_features(tweet)
     # print(personality_detection(tweet))
     # print(str(features_list["Average_Word_Length"]))
@@ -380,7 +381,7 @@ def greet(tweet):
         label = "Non Hate"
     
 
-    return label,str(prediction)+"%",str(1-prediction)+"%"
+    return label,str(prediction_value)+"%",str(1-prediction_value)+"%"
 
 # demo = gr.Interface(fn=greet, inputs="text", outputs="text")
 demo = gr.Interface(