Update main.py

main.py

@@ -123,6 +123,14 @@ def get_prompts(prompt_text):
     else:
         return text_gen("", timestamp)
 
+def initialize_api_connection():
+    global headers
+    API_TOKEN = os.getenv("HF_READ_TOKEN") # it is free
+    headers = {"Authorization": f"Bearer {API_TOKEN}"}
+
+# Run initialization functions on startup
+initialize_api_connection()
+
 @app.get("/generate_prompts")
 def generate_prompts(prompt_text: str):
     return get_prompts(prompt_text)