ModelSmith-AI / backend /core /deployment_generator.py
ACA050's picture
Upload 79 files
a309487 verified
class DeploymentGenerator:
def generate_fastapi_app(self, model_path):
template = f'''
from fastapi import FastAPI
import joblib
import pandas as pd
app = FastAPI()
model = joblib.load("{model_path}")
@app.post("/predict")
async def predict(data: dict):
df = pd.DataFrame([data])
preds = model.predict(df)
return {{"prediction": preds.tolist()}}
'''
return template
def generate_dockerfile(self):
return '''
FROM python:3.9
WORKDIR /app
COPY . /app
RUN pip install fastapi uvicorn pandas scikit-learn joblib
CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]
'''