Spaces:
Sleeping
Sleeping
from fastapi import FastAPI, HTTPException,APIRouter,Request | |
from fastapi.responses import HTMLResponse | |
from fastapi.staticfiles import StaticFiles | |
from starlette.responses import FileResponse | |
from fastapi.middleware.cors import CORSMiddleware | |
from transformers import pipeline | |
from pydantic import BaseModel | |
class UserRequest(BaseModel): | |
text: str | |
length: int | |
app = FastAPI() | |
origins = ["*"] | |
app.add_middleware( | |
CORSMiddleware, | |
allow_origins=origins, | |
) | |
app.mount("/files/", StaticFiles(directory='../code'), name="index") | |
async def read_index(): | |
return FileResponse('app/html/index.html') | |
async def get_summary(post: UserRequest): | |
summarizer = pipeline("summarization", model="Falconsai/text_summarization") | |
query=str(post.text) | |
query_clean=""+query | |
summary = summarizer(query_clean, max_length=post.length, min_length=10, do_sample=False) | |
return (summary[0]['summary_text']) | |
async def converter(temperature:float): | |
return (temperature*(9/5))+32 | |