Chris4K commited on
Commit
689088d
·
verified ·
1 Parent(s): c118719

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -45,9 +45,9 @@ login(token=hf_token)
45
  # Define the model pipeline with additional generation parameters
46
  model_pipeline = pipeline(
47
  model="meta-llama/Llama-3.2-1B",
48
- use_auth_token=hf_token,
49
- max_length=250, # You can increase this if needed
50
- max_new_tokens=100 # Limit how many tokens are generated
51
  )
52
 
53
  # Use the pipeline in HuggingFacePipeline
 
45
  # Define the model pipeline with additional generation parameters
46
  model_pipeline = pipeline(
47
  model="meta-llama/Llama-3.2-1B",
48
+ #use_auth_token=hf_token,
49
+ max_length=550, # You can increase this if needed
50
+ max_new_tokens=200 # Limit how many tokens are generated
51
  )
52
 
53
  # Use the pipeline in HuggingFacePipeline