bart summarizer

main.py

@@ -4,7 +4,9 @@ from fastapi import FastAPI
 from pydantic import BaseModel
 import re
 import os
+from transformers import pipeline
 
+bart_summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
 
 
 app = FastAPI()
@@ -105,21 +107,7 @@ async def root(item: Item):
 
         article = extract_article_content(item.url)
 
-        token = os.environ.get("huggingface_token")
-        if token:
-            print(token)
-        API_URL = "https://api-inference.huggingface.co/models/facebook/bart-large-cnn"
-        headers = {'Authorization': token}
-
-            
-        params = {'do_sample': False}
-
-        response = requests.post(API_URL, headers=headers, json={
-            'inputs': article,
-            'parameters': params
-        })
-
-        output = response.json()
+        output = summarizer(ARTICLE, max_length=130, min_length=30, do_sample=False)
     
         return {"summary": output}