Update app.py

app.py

@@ -7,6 +7,11 @@ from io import StringIO
 import os
 from flask import Flask, request, jsonify
 
+# Accessing the token from environment variables
+hf_token = os.getenv('HF_API_TOKEN')
+
+# Ensure that you load the token before making API requests
+
 # Set environment variable to avoid floating-point errors
 os.environ['TF_ENABLE_ONEDNN_OPTS'] = '0'
 
@@ -68,7 +73,7 @@ def generate_synthetic_data(description, columns):
     try:
         formatted_prompt = format_prompt(description, columns)
         payload = {"inputs": formatted_prompt, "parameters": generation_params}
-        headers = {"Authorization": f"Bearer {HfFolder.get_token()}"}
+        headers = {"Authorization": f"Bearer {hf_token}"}
         response = requests.post(API_URL, headers=headers, json=payload)
         
         if response.status_code == 200: