Spaces:
Sleeping
Sleeping
File size: 1,099 Bytes
f7b3a0a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 |
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")
@app.get("/")
async def read_index():
return FileResponse('app/html/index.html')
@app.post("/Summarizer/")
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'])
@app.post("/Freedom Units/")
async def converter(temperature:float):
return (temperature*(9/5))+32
|