matches / app /app.py
Joseph Ibochi
update:model mod for handling NAN
0b6fe9c
raw
history blame contribute delete
703 Bytes
from fastapi import FastAPI
from pydantic import BaseModel
from typing import Dict, List
from app.model import RoommateMatcher
app = FastAPI(
title="Roommate Matcher API",
openapi_url="/openapi.json", # πŸ‘ˆ expose OpenAPI schema
docs_url="/docs", # πŸ‘ˆ enable Swagger docs
redoc_url=None # optional
)
matcher = RoommateMatcher()
class MatchRequest(BaseModel):
current_user: Dict
other_users: List[Dict]
@app.post("/match")
def match(request: MatchRequest):
try:
result = matcher.predict(request.current_user, request.other_users)
return {"matches": result}
except Exception as e:
return {"error": str(e)}