Accke commited on
Commit
8a958af
1 Parent(s): 714fe7f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -3
app.py CHANGED
@@ -7,7 +7,7 @@ import os
7
  from fastapi import FastAPI, HTTPException
8
  from pydantic import BaseModel
9
  from typing import List
10
- from tensorflow.keras.models import load_model
11
  from sklearn.preprocessing import StandardScaler
12
 
13
  class InputData(BaseModel):
@@ -15,8 +15,13 @@ class InputData(BaseModel):
15
 
16
  app = FastAPI()
17
 
18
- # Cargar el modelo entrenado
19
- model = load_model('modelo_entrenado.h5')
 
 
 
 
 
20
 
21
  # Ruta de predicción
22
  @app.post("/predict/")
 
7
  from fastapi import FastAPI, HTTPException
8
  from pydantic import BaseModel
9
  from typing import List
10
+ from tensorflow.keras.models import model_from_json
11
  from sklearn.preprocessing import StandardScaler
12
 
13
  class InputData(BaseModel):
 
15
 
16
  app = FastAPI()
17
 
18
+ # Cargar el modelo desde JSON
19
+ with open('model.json', 'r') as json_file:
20
+ model_json = json_file.read()
21
+ model = model_from_json(model_json)
22
+
23
+ # Cargar los pesos en el modelo
24
+ model.load_weights('model_weights.h5')
25
 
26
  # Ruta de predicción
27
  @app.post("/predict/")