Remove torch.sigmoid

app.py

@@ -372,7 +372,7 @@ def greet(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()
+    predicted_class = torch.round(prediction).item()
     print("end")
     if (predicted_class==0.0):
         label = "Hate"