YaserDS-777 commited on
Commit
1098c9d
1 Parent(s): f7bed26

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ if not api_token:
10
  raise ValueError("API token is not set. Please set the HUGGINGFACE_API_TOKEN environment variable.")
11
 
12
  # Initialize the text generation pipeline with authentication
13
- pipe = pipeline("text-generation", model="meta-llama/Meta-Llama-3.1-405B", use_auth_token=api_token)
14
 
15
  # Define the function to generate text
16
  def generate_text(prompt):
 
10
  raise ValueError("API token is not set. Please set the HUGGINGFACE_API_TOKEN environment variable.")
11
 
12
  # Initialize the text generation pipeline with authentication
13
+ pipe = pipeline("text-generation", model="meta-llama/Meta-Llama-3.1-405B", use_auth_token=True)
14
 
15
  # Define the function to generate text
16
  def generate_text(prompt):