Spaces:
Sleeping
Sleeping
amaye15
commited on
Commit
·
0d22171
1
Parent(s):
db08d02
Debug - Main page
Browse files- 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(
|