Update app.py

app.py

@@ -57,7 +57,7 @@ def predict_siniestros(area, traf, vage, tipov, valorv, edad, Gen):
     df = pd.DataFrame.from_dict({'area': [area], 'traffic_indexe': [traf], 'veh_age': [vage], 'veh_body': [tipov], 'veh_value': [valorv], 'Edad': [edad], 'gender_M': [Gen]})
     
     pred=tree.predict(df)
-    return {'Es posible que tengas que reclamar al seguro': float(pred[0])}
+    return {'Es posible que tengas que reclamar al seguro': int(pred[0])}
     
 area = gr.inputs.Radio(['1', '2', '3', '4', '5', '6'], label="Area: del 1 al 6")
 traf = gr.inputs.Slider(minimum=0, maximum=200, default=100, label="Indice de trafico")