subbunanepalli commited on
Commit
add4c06
·
verified ·
1 Parent(s): b432eac

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -93,7 +93,7 @@ class PredictionRequest(BaseModel):
93
 
94
  @app.get("/")
95
  def root():
96
- return {"status": "healthy", "message": "LogReg TF-IDF API is running"}
97
 
98
  @app.post("/validate")
99
  def validate_input(request: PredictionRequest):
@@ -105,7 +105,7 @@ def validate_input(request: PredictionRequest):
105
 
106
  @app.post("/predict")
107
  def predict(request: PredictionRequest):
108
- try:
109
  input_data = pd.DataFrame([request.transaction_data.dict()])
110
 
111
  text_input = f"""
@@ -171,5 +171,5 @@ def predict(request: PredictionRequest):
171
 
172
  return response
173
 
174
- except Exception as e:
175
- raise HTTPException(status_code=500, detail=str(e))
 
93
 
94
  @app.get("/")
95
  def root():
96
+ return {"status": "healthy", "message": "XGBoost TF-IDF API is running"}
97
 
98
  @app.post("/validate")
99
  def validate_input(request: PredictionRequest):
 
105
 
106
  @app.post("/predict")
107
  def predict(request: PredictionRequest):
108
+ try:
109
  input_data = pd.DataFrame([request.transaction_data.dict()])
110
 
111
  text_input = f"""
 
171
 
172
  return response
173
 
174
+ except Exception as e:
175
+ raise HTTPException(status_code=500, detail=str(e))