lpedro commited on
Commit
42d6166
·
verified ·
1 Parent(s): 5edd301

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +7 -1
main.py CHANGED
@@ -1,10 +1,16 @@
1
  from typing import List
2
  from fastapi import FastAPI
 
3
 
4
  from models import RequestModel, ResponseModel
5
  from similarity import check_similarity
6
 
7
- app = FastAPI()
 
 
 
 
 
8
 
9
 
10
  @app.post("/similarity", summary="Perform images similarity", response_model=List[ResponseModel], tags=["Similarities"])
 
1
  from typing import List
2
  from fastapi import FastAPI
3
+ from fastapi.responses import RedirectResponse
4
 
5
  from models import RequestModel, ResponseModel
6
  from similarity import check_similarity
7
 
8
+ app = FastAPI(title="Prozis Similarity Service", version="1.0.0")
9
+
10
+
11
+ @app.get("/", include_in_schema=False)
12
+ async def root():
13
+ return RedirectResponse(url="/docs")
14
 
15
 
16
  @app.post("/similarity", summary="Perform images similarity", response_model=List[ResponseModel], tags=["Similarities"])