Update app.py

app.py

@@ -187,7 +187,12 @@ def decrypt_prediction(user_id):
     fhe_api.generate_private_and_evaluation_keys(force=False)
 
     predictions = fhe_api.deserialize_decrypt_dequantize(encrypted_prediction)
-    return predictions
+
+    if(predictions[0][0] >= 0.5){
+        return "Safe file"
+    }else{
+        return "Malware"
+    }
 
 
 if __name__ == "__main__":