neural-news-lab's picture
Update app.py
e1c318f verified
raw
history blame contribute delete
673 Bytes
from fastapi import FastAPI
from pydantic import BaseModel
# импорт твоей логики
from new_sum import generate_summary
# ======================
# APP INIT
# ======================
app = FastAPI()
@app.get("/")
def root():
return {"status": "ok"}
@app.get("/health")
def health():
return {"model": "loaded"}
# ======================
# REQUEST SCHEMA
# ======================
class NewsRequest(BaseModel):
text: str
# ======================
# API ENDPOINT
# ======================
@app.post("/summarize")
def summarize(req: NewsRequest):
summary = generate_summary(req.text)
return {"summary": summary}