Spaces:

facuvillegas
/

prueba_eae

Sleeping

facuvillegas commited on Mar 7, 2025

Commit

6987254

verified ·

1 Parent(s): bb38e54

predict_proba

Files changed (1) hide show

app.py CHANGED Viewed

@@ -25,5 +25,6 @@ async def root():
 @app.post("/predecir")
 async def predecir(datos: InputData):
     X = np.array(datos.input).reshape(1, -1)
-    prediccion = modelo.predict(X)
-    return {"prediccion": int(prediccion[0])}

 @app.post("/predecir")
 async def predecir(datos: InputData):
     X = np.array(datos.input).reshape(1, -1)
+    probabilidades = modelo.predict_proba(X)
+    # prediccion = modelo.predict(X)
+    return {"probs": probabilidades[0].tolist()}