Spaces:
Sleeping
Sleeping
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)
|