rasmodev commited on
Commit
4bc5036
·
1 Parent(s): 6fb9745

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +3 -3
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)