Update app.py

app.py

@@ -9,6 +9,8 @@ import openai
 import warnings
 warnings.filterwarnings("ignore")
 
+api_token = os.getenv('H_TOKEN')
+
 # Image-to-text
 def img2txt(url):
     print("Initializing captioning model...")
@@ -23,7 +25,7 @@ def img2txt(url):
 # Text-to-story
 def txt2story(img_text, top_k, top_p, temperature):
 
-    headers = {"Authorization": f"Bearer {os.environ['H_TOKEN']}"}
+    headers = {"Authorization": f"Bearer {api_token}"}
 
     data = {
         "model": "togethercomputer/llama-2-70b-chat",
@@ -47,7 +49,7 @@ def txt2story(img_text, top_k, top_p, temperature):
 def txt2speech(text):
     print("Initializing text-to-speech conversion...")
     API_URL = "https://api-inference.huggingface.co/models/espnet/kan-bayashi_ljspeech_vits"
-    headers = {"Authorization": f"Bearer {os.environ['H_TOKEN']}"}
+    headers = {"Authorization": f"Bearer {api_token }"}
     payloads = {'inputs': text}
 
     response = requests.post(API_URL, headers=headers, json=payloads)