FinTech / main.py
tensorgirl's picture
Update main.py
69b5a42 verified
from fastapi import FastAPI
from pydantic import BaseModel, validator
from utils import summary
import os
from huggingface_hub import login
os.environ['HF_HOME'] = '/hug/cache/'
os.environ['TRANSFORMERS_CACHE'] = '/blabla/cache/'
class Item(BaseModel):
FileURL: str = "https://www.bseindia.com/stockinfo/AnnPdfOpen.aspx?Pname=d141ef4f-7856-4236-8f6f-efe09592df40.pdf"
memo: str = "Please find attached RTA Certificate u/r 74(5) of SEBI (DP) Regulations 2018 for QE March 2024"
TypeofAnnouncement: str = "General_Announcements"
Descriptor: str = "Certificate under Reg. 74 (5) of SEBI (DP) Regulations 2018"
caption: str = "Compliances-Certificate under Reg. 74 (5) of SEBI (DP) Regulations 2018"
newsdate: str = "2024-04-08T13:05:27"
symbol: str = "EDELWEISS"
app = FastAPI()
@app.get("/")
async def root():
return {"Summarize":"Version 1.5 'Images Added'"}
@app.post("/Summarize/")
def read_user(input_json: Item):
return summary(input_json.dict())