Update app.py

app.py

@@ -3,6 +3,7 @@ import pandas as pd
 import numpy as np
 from modelo_risco.modelo_risco_coletivo import *
 from util import string_to_list_int, string_to_list_float
+import math
 
 st.title('Modelo de Risco Coletivo')
 
@@ -13,7 +14,7 @@ pn = string_to_list_float(st.text_input('Probabilidades da distribuição Frequ
 
 
 if len(x) == len(px) and len(n) == len(pn):
-    if (abs(round(sum(px)) - 1.0) >= .001) or (abs(round(sum(pn)) - 1.0) >= .001):
+    if not math.isclose(sum(px), 1.0) or not math.isclose(sum(pn), 1.0):
         st.warning('A soma das probabilidades precisa ser igual a 1.0')       
          
     else: