Practice_API / app /main.py
Sankie005's picture
Initial Commit
f7b3a0a verified
raw
history blame contribute delete
No virus
1.1 kB
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