Create main.py

main.py

@@ -0,0 +1,21 @@
+from fastapi import FastAPI
+from pydantic import BaseModel
+import numpy as np
+from tensorflow.keras.models import load_model
+
+app = FastAPI()
+model = load_model("titanic_model.h5")
+
+class InputData(BaseModel):
+    pclass: int
+    sex: str
+    age: float
+    fare: float
+
+@app.post("/predict")
+async def predict(data: InputData):
+    sex_num = 1 if data.sex.lower() == "male" else 0
+    input_array = np.array([[data.pclass, sex_num, data.age, data.fare]])
+    prediction = model.predict(input_array)[0][0]
+    result = "Sobrevivió" if prediction > 0.5 else "No sobrevivió"
+    return {"data": [result]}