amaye15 commited on
Commit
0d22171
·
1 Parent(s): db08d02

Debug - Main page

Browse files
Files changed (1) hide show
  1. api/main.py +5 -0
api/main.py CHANGED
@@ -202,6 +202,11 @@ def get_model():
202
  return tuned_model
203
 
204
 
 
 
 
 
 
205
  # Define the prediction endpoint
206
  @app.post("/predict", response_model=PredictionResponse)
207
  async def predict(
 
202
  return tuned_model
203
 
204
 
205
+ @app.get("/")
206
+ async def root():
207
+ return RedirectResponse(url="/docs")
208
+
209
+
210
  # Define the prediction endpoint
211
  @app.post("/predict", response_model=PredictionResponse)
212
  async def predict(