Arafath10 commited on
Commit
6cb03a2
·
verified ·
1 Parent(s): e3d2180

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +4 -4
main.py CHANGED
@@ -178,8 +178,8 @@ async def your_continuous_function(page: int,paginate: int,Tenant: str):
178
  @app.get("/get_latest_model_updated_time")
179
  async def model_updated_time():
180
  try:
181
- m_time_encoder = os.path.getmtime('encoders.joblib')
182
- m_time_model = os.path.getmtime('xgb_model.joblib')
183
  return {"base model created time ":datetime.datetime.fromtimestamp(m_time_encoder),
184
  "last model updated time":datetime.datetime.fromtimestamp(m_time_model)}
185
  except:
@@ -204,8 +204,8 @@ def predict(customer_name: str,
204
 
205
  try:
206
  # Load your trained model and encoders
207
- xgb_model = load('xgb_model.joblib')
208
- encoders = load('encoders.joblib')
209
  except:
210
  return {"no model found so first trained the model using data fecther"}
211
 
 
178
  @app.get("/get_latest_model_updated_time")
179
  async def model_updated_time():
180
  try:
181
+ m_time_encoder = os.path.getmtime('model/curfox_encoders.joblib')
182
+ m_time_model = os.path.getmtime('model/curfox_xgb_model.joblib')
183
  return {"base model created time ":datetime.datetime.fromtimestamp(m_time_encoder),
184
  "last model updated time":datetime.datetime.fromtimestamp(m_time_model)}
185
  except:
 
204
 
205
  try:
206
  # Load your trained model and encoders
207
+ xgb_model = load('model/curfox_xgb_model.joblib')
208
+ encoders = load('model/curfox_encoders.joblib')
209
  except:
210
  return {"no model found so first trained the model using data fecther"}
211