Spaces:
Sleeping
Sleeping
AndresAlegre
commited on
Commit
•
44d4e7a
1
Parent(s):
b6861a1
Update app.py
Browse files
app.py
CHANGED
@@ -97,7 +97,7 @@ def predict(input: HeartDiseaseInput):
|
|
97 |
predicted_class_python = predicted_class[0][0].item()
|
98 |
|
99 |
# Interpretar la predicción
|
100 |
-
result = "
|
101 |
|
102 |
return {"prediction": result, "probability": prediction_prob_python}
|
103 |
except Exception as e:
|
|
|
97 |
predicted_class_python = predicted_class[0][0].item()
|
98 |
|
99 |
# Interpretar la predicción
|
100 |
+
result = "Si tiene ataques cardiacos" if predicted_class_python == 1 else "No tiene ataques cardíacos"
|
101 |
|
102 |
return {"prediction": result, "probability": prediction_prob_python}
|
103 |
except Exception as e:
|