Spaces:
Runtime error
Runtime error
update app
Browse files
app.py
CHANGED
@@ -25,7 +25,7 @@ def generate(gen_prompt, max_tokens, temperature=0.6, seed=42):
|
|
25 |
output = model.generate(input_ids=input_ids, do_sample=True, top_p=0.95, temperature=temperature, max_length=max_length)
|
26 |
generated_text = tokenizer.decode(output.flatten())
|
27 |
if generated_text.startswith(BOS):
|
28 |
-
generated_text =
|
29 |
return generated_text
|
30 |
|
31 |
iface = gr.Interface(
|
|
|
25 |
output = model.generate(input_ids=input_ids, do_sample=True, top_p=0.95, temperature=temperature, max_length=max_length)
|
26 |
generated_text = tokenizer.decode(output.flatten())
|
27 |
if generated_text.startswith(BOS):
|
28 |
+
generated_text = generated_text[len(BOS):]
|
29 |
return generated_text
|
30 |
|
31 |
iface = gr.Interface(
|