RaulHuaroteZegarra commited on
Commit
29c6739
1 Parent(s): ab85069

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -3
app.py CHANGED
@@ -3,7 +3,7 @@ from fastapi import FastAPI, HTTPException
3
  from pydantic import BaseModel
4
  import numpy as np
5
  from typing import List
6
- import pickle
7
 
8
  class InputData(BaseModel):
9
  data: List[float] # Lista de características numéricas (flotantes)
@@ -12,8 +12,12 @@ app = FastAPI()
12
 
13
  # Función para construir el modelo manualmente
14
  def build_model():
15
- with open("clf_train.pkl", "rb") as tf:
16
- miarbol = pickle.load(tf)
 
 
 
 
17
  return miarbol
18
 
19
  model = build_model() # Construir el modelo al iniciar la aplicación
 
3
  from pydantic import BaseModel
4
  import numpy as np
5
  from typing import List
6
+ #import pickle
7
 
8
  class InputData(BaseModel):
9
  data: List[float] # Lista de características numéricas (flotantes)
 
12
 
13
  # Función para construir el modelo manualmente
14
  def build_model():
15
+ from pickle import load
16
+ with open("clf_train.pkl", "rb") as f:
17
+ miarbol = load(f)
18
+
19
+ #with open("clf_train.pkl", "rb") as tf:
20
+ # miarbol = pickle.load(tf)
21
  return miarbol
22
 
23
  model = build_model() # Construir el modelo al iniciar la aplicación