Update app.py

app.py

@@ -191,10 +191,10 @@ def decrypt_prediction(user_id):
 
     predictions = fhe_api.deserialize_decrypt_dequantize(encrypted_prediction)
 
-    if(predictions[0][0] >= 0.5):
-        return "Safe file", predictions
+    if(predictions[0][1] >= 0.5):
+        return "Safe file with class probabilities: ", predictions
     else:
-        return "Malware", predictions
+        return "Malware file with class probabilities: ", predictions