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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -46,8 +46,8 @@ login(token=hf_token)
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
 
46
  model_pipeline = pipeline(
47
  model="meta-llama/Llama-3.2-1B",
48
  #use_auth_token=hf_token,
49
+ max_length=1000, # You can increase this if needed
50
+ max_new_tokens=500 # Limit how many tokens are generated
51
  )
52
 
53
  # Use the pipeline in HuggingFacePipeline