Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
)
|
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)
|