Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
16 |
-
|
|
|
|
|
|
|
|
|
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
|