similarity / main.py
ricardo238costa's picture
Create main.py
9c36111 verified
raw
history blame contribute delete
398 Bytes
from typing import List
from fastapi import FastAPI
from models import RequestModel, ResponseModel
from similarity import check_similarity
app = FastAPI()
@app.post("/similarity", summary="Perform images similarity", response_model=List[ResponseModel], tags=["Similarities"])
async def similarity(images: List[RequestModel]):
similarities = check_similarity(images)
return similarities