Dhahlan2000 commited on
Commit
8c59ba3
·
1 Parent(s): 20d39a9

added max new tokens

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -47,7 +47,7 @@ model = AutoModelForCausalLM.from_pretrained(
47
  torch_dtype=torch.bfloat16,
48
  token = access_token
49
  )
50
- pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
51
 
52
  def validate_ticker(ticker):
53
  # Ensure ticker is uppercase and length is reasonable (1-5 characters)
 
47
  torch_dtype=torch.bfloat16,
48
  token = access_token
49
  )
50
+ pipe = pipeline("text-generation", model=model, tokenizer=tokenizer, max_new_tokens=1048)
51
 
52
  def validate_ticker(ticker):
53
  # Ensure ticker is uppercase and length is reasonable (1-5 characters)