Update main.py

main.py

@@ -16,16 +16,17 @@ app = Flask(__name__)
 
 import os
 
-# Replace 'YOUR_API_KEY' with your actual API key
-os.environ['API_KEY'] = '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'
+
+os.environ['API_URL'] = 'https://api-inference.huggingface.co/models/ProsusAI/finbert'
 
 # Use os.getenv('API_KEY') to access the value
 api_key = os.getenv('API_KEY')
+api_url = os.getenv('API_URL')
 
 headers = {"Authorization": f"Bearer {api_key}"}
-API_URL = "https://api-inference.huggingface.co/models/ProsusAI/finbert"
+# API_URL = "https://api-inference.huggingface.co/models/ProsusAI/finbert"
 def query(payload):
-    response = requests.post(API_URL, headers=headers, json=payload)
+    response = requests.post(api_url, headers=headers, json=payload)
     return response.json()