Update app.py

app.py

@@ -7,6 +7,7 @@ import json
 import time
 
 API_URL = os.environ.get("API_URL")
+TOKEN = os.environ.get("TOKEN")
 supported_languages = ['zh', 'en']
 
 output_dir = 'outputs'
@@ -103,7 +104,12 @@ def predict(prompt, style, audio_file_pth, agree):
     
     start = time.time()
     # Send the data as a POST request
-    response = requests.post(API_URL, json=data, timeout=60)
+
+    headers = {
+        "Authorization": f"Bearer {TOKEN}"
+    }
+    
+    response = requests.post(API_URL, json=data, headers=headers, timeout=60)
     print(f'Get response successfully within {time.time() - start}')
 
     # Check the response