API inicio

Author: SergioPardo55

DataModel.py

@@ -0,0 +1,8 @@
+from pydantic import BaseModel
+
+class DataModel(BaseModel):
+    review_es : str
+    sentimiento : str
+
+    def columns(self):
+        return ['review_es', 'sentimiento']

PredictionModel.py

@@ -0,0 +1,10 @@
+from joblib import load
+
+class Model:
+
+    def __init__(self,columns):
+        self.model = load("assets/pipelineNaiveBayes.joblib")
+
+    def make_predictions(self, data):
+        result = self.model.predict(data)
+        return result

__pycache__/DataModel.cpython-310.pyc

@@ -0,0 +1,28 @@
+from typing import Optional
+
+from fastapi import FastAPI
+
+from DataModel import DataModel
+
+from joblib import load
+
+import pandas as pd
+
+app = FastAPI()
+
+
+@app.get("/")
+def read_root():
+   return {"Hello": "World"}
+
+@app.get("/items/{item_id}")
+def read_item(item_id: int, q: Optional[str] = None):
+   return {"item_id": item_id, "q": q}
+
+@app.post("/predict")
+def make_predictions(dataModel: DataModel):
+    df = pd.DataFrame(dataModel.dict(), columns=dataModel.dict().keys(), index=[0])
+    df.columns = dataModel.columns()
+    model = load("assets/pipelineNaiveBayes.joblib")
+    result = model.predict(df)
+    return result