bobpopboom commited on
Commit
146ed17
·
verified ·
1 Parent(s): c215361

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -32,7 +32,7 @@ def generate_text_streaming(prompt, max_new_tokens=128):
32
  inputs.input_ids.tolist(), # ctranslate2 needs list of token ids
33
  max_length=1, # Generate one token at a time
34
  beam_size=1, # Greedy decoding
35
- ).eval()
36
 
37
  new_token_id = outputs[0][0][-1] # Extract the generated token ID
38
  new_token = tokenizer.decode(new_token_id, skip_special_tokens=True)
 
32
  inputs.input_ids.tolist(), # ctranslate2 needs list of token ids
33
  max_length=1, # Generate one token at a time
34
  beam_size=1, # Greedy decoding
35
+ )
36
 
37
  new_token_id = outputs[0][0][-1] # Extract the generated token ID
38
  new_token = tokenizer.decode(new_token_id, skip_special_tokens=True)