add handle of pipeline summarization

app/api.py

@@ -1,5 +1,5 @@
 import config.config as config
-from app.src.src import pipeline_sentiment, pipeline_stats
+from app.src.src import pipeline_sentiment, pipeline_stats, pipeline_summarize
 
 from fastapi import FastAPI
 from pydantic import BaseModel
@@ -9,6 +9,8 @@ import pandas as pd
 import os
 
 sentiment_model = pipeline(model=config.sentiment_model)
+sum_model = pipeline(model=config.sum_model, use_fast=True)
+
 app = FastAPI()
 
 class YouTubeUrl(BaseModel):
@@ -30,6 +32,12 @@ def get_stats_sent():
         data = pd.read_csv(f"{config.DATA_FILE}")
         return pipeline_stats(data)
 
+@app.get('/summarization')
+def get_summarize():
+    if f"{config.NAME_DATA}" in os.listdir(f"{config.PATH_DATA}"):
+        data = pd.read_csv(f"{config.DATA_FILE}")
+        return pipeline_summarize(data['text_comment'], sum_model)
+
 
 if __name__ == '__main__':
     uvicorn.run(app, host='127.0.0.1', port=80)