update api.py

app/api.py

@@ -1,4 +1,4 @@
-from config import config, secrets
+from config import config
 from app.src.src import pipeline_sentiment, pipeline_stats, pipeline_summarize
 
 from fastapi import FastAPI
@@ -10,7 +10,7 @@ import os
 
 # sentiment_model = pipeline(model=config.sentiment_model)
 # sum_model = pipeline(model=config.sum_model, use_fast=True)
-headers = {"Authorization": f"Bearer {secrets.API_TOKEN}"}
+headers = {"Authorization": f"Bearer {os.environ.get('API_TOKEN')}"}
 SENT_API_URL = f"https://api-inference.huggingface.co/models/{config.sentiment_model}"
 SUM_API_URL = f"https://api-inference.huggingface.co/models/{config.sum_model}"
 
@@ -25,7 +25,7 @@ def read_root():
 
 @app.post('/comments')
 def get_comments(url_video: YouTubeUrl):
-    data = pipeline_sentiment(url_video.url_video, config.API_KEY, headers, SENT_API_URL)
+    data = pipeline_sentiment(url_video.url_video, os.environ.get("API_KEY"), headers, SENT_API_URL)
     data.to_csv(f"{config.DATA_FILE}", index=False)
     return data # {'message': 'Success'}