Update main.py

main.py

@@ -8,13 +8,21 @@ import re
 import numpy as np
 import pandas as pd
 from collections import OrderedDict
+import os
 # import requests
 # from bs4 import BeautifulSoup
 
 app = Flask(__name__)
 
+import os
 
-headers = {"Authorization": f"Bearer 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"}
+# Replace 'YOUR_API_KEY' with your actual API key
+os.environ['API_KEY'] = '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'
+
+# Use os.getenv('API_KEY') to access the value
+api_key = os.getenv('API_KEY')
+
+headers = {"Authorization": f"Bearer {api_key}"}
 API_URL = "https://api-inference.huggingface.co/models/ProsusAI/finbert"
 def query(payload):
     response = requests.post(API_URL, headers=headers, json=payload)