Update app.py

app.py

@@ -6,6 +6,10 @@ from huggingface_hub import HfFolder, login
 from io import StringIO
 import os
 from flask import Flask, request, jsonify
+from huggingface_hub import HfFolder
+
+print(HfFolder.get_token())
+login()
 
 # Load GPT-2 model and tokenizer
 
@@ -65,6 +69,7 @@ def format_prompt(description, columns):
 
 API_URL = "https://api-inference.huggingface.co/models/meta-llama/Meta-Llama-3.1-8B"
 
+
 generation_params = {
     "top_p": 0.90,
     "temperature": 0.8,
@@ -86,6 +91,7 @@ def generate_synthetic_data(description, columns):
         else:
             raise ValueError("Unexpected response format or missing 'generated_text' key")
     else:
+        print(f"Error details: {response.text}") 
         raise ValueError(f"API request failed with status code {response.status_code}: {response.text}")
 
 def generate_large_synthetic_data(description, columns, num_rows=1000, rows_per_generation=100):