Update main.py
Browse files
main.py
CHANGED
|
@@ -1,7 +1,8 @@
|
|
|
|
|
|
|
|
| 1 |
from fastapi import FastAPI
|
|
|
|
| 2 |
from pydantic import BaseModel
|
| 3 |
-
import pickle
|
| 4 |
-
import pandas as pd
|
| 5 |
|
| 6 |
# Load the saved model
|
| 7 |
with open("model_and_key_components.pkl", "rb") as f:
|
|
@@ -66,5 +67,4 @@ async def predict_income(data: IncomePredictionRequest):
|
|
| 66 |
raise
|
| 67 |
|
| 68 |
if __name__ == "__main__":
|
| 69 |
-
import uvicorn
|
| 70 |
uvicorn.run(app, host="0.0.0.0", port=7860)
|
|
|
|
| 1 |
+
import pandas as pd
|
| 2 |
+
import pickle
|
| 3 |
from fastapi import FastAPI
|
| 4 |
+
import uvicorn
|
| 5 |
from pydantic import BaseModel
|
|
|
|
|
|
|
| 6 |
|
| 7 |
# Load the saved model
|
| 8 |
with open("model_and_key_components.pkl", "rb") as f:
|
|
|
|
| 67 |
raise
|
| 68 |
|
| 69 |
if __name__ == "__main__":
|
|
|
|
| 70 |
uvicorn.run(app, host="0.0.0.0", port=7860)
|