WICKED4950 commited on
Commit
d273c52
·
verified ·
1 Parent(s): f77535c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ tokenizer.pad_token = tokenizer.eos_token
16
  with strategy.scope():
17
  model = TFAutoModelForCausalLM.from_pretrained(name)
18
 
19
- def raw_pred(input, model, tokenizer, max_length=50, temperature=0.2):
20
  input_ids = tokenizer.encode(input, return_tensors='tf')
21
 
22
  # Initialize variables
 
16
  with strategy.scope():
17
  model = TFAutoModelForCausalLM.from_pretrained(name)
18
 
19
+ def raw_pred(input, model, tokenizer, max_length=1024, temperature=0.2):
20
  input_ids = tokenizer.encode(input, return_tensors='tf')
21
 
22
  # Initialize variables