Update app.py

app.py

@@ -1,11 +1,12 @@
 import os
-
 import gradio as gr
 from transformers import pipeline
-HF_TOKEN = os.environ.get("llama3", None)
 
-# Initialize the text generation pipeline
-pipe = pipeline("text-generation", model="meta-llama/Meta-Llama-3.1-405B")
+# Get the Hugging Face API token from environment variables
+api_token = os.getenv("HUGGINGFACE_API_TOKEN")
+
+# Initialize the text generation pipeline with authentication
+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):