Update app/main.py

app/main.py

@@ -5,6 +5,8 @@ from pydantic import BaseModel
 import re
 import os
 
+token = os.environ.get("huggingface_token")
+key = os.environ.get("rapid_key")
 
 app = FastAPI()
 
@@ -52,7 +54,7 @@ async def root(item: Item):
                    }
         headers = {
             "content-type": "application/json",
-            "X-RapidAPI-Key": os.environ.get("rapid_key"),
+            "X-RapidAPI-Key": key,
             "X-RapidAPI-Host": "text-analysis12.p.rapidapi.com"
         }
 
@@ -104,8 +106,10 @@ async def root(item: Item):
 
         article = extract_article_content(item.url)
 
+        
+
         API_URL = "https://api-inference.huggingface.co/models/facebook/bart-large-cnn"
-        headers = {'Authorization': os.environ.get("huggingface_token")}
+        headers = {'Authorization': token}
 
 
         def query(payload):