Update app.py

app.py

@@ -3,10 +3,16 @@ import gradio as gr
 from transformers import pipeline
 
 # Get the Hugging Face API token from environment variables
-HUGGINGFACE_API_TOKEN = os.getenv("HUGGINGFACE_API_TOKEN")
+api_token = os.getenv("HUGGINGFACE_API_TOKEN")
+
+# Check if the API token is set
+
+print('-----------------',api_token,'-------------')
+if not api_token:
+    raise ValueError("API token is not set. Please set the HUGGINGFACE_API_TOKEN environment variable.")
 
 # Initialize the text generation pipeline with authentication
-pipe = pipeline("text-generation", model="meta-llama/Meta-Llama-3.1-405B", use_auth_token=HUGGINGFACE_API_TOKEN)
+pipe = pipeline("text-generation", model="meta-llama/Meta-Llama-3.1-405B", use_auth_token=api_token)
 
 # Define the function to generate text
 def generate_text(prompt):