Spaces:
Paused
Paused
Carlos Rosas
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -88,6 +88,7 @@ class CassandreChatBot:
|
|
88 |
eos_token_id=tokenizer.eos_token_id
|
89 |
)
|
90 |
|
|
|
91 |
generated_text = tokenizer.decode(output[0])
|
92 |
generated_text = '<h2 style="text-align:center">Réponse</h3>\n<div class="generation">' + format_references(generated_text) + "</div>"
|
93 |
fiches_html = '<h2 style="text-align:center">Sources</h3>\n' + fiches_html
|
|
|
88 |
eos_token_id=tokenizer.eos_token_id
|
89 |
)
|
90 |
|
91 |
+
# Just decode and display the generated text directly
|
92 |
generated_text = tokenizer.decode(output[0])
|
93 |
generated_text = '<h2 style="text-align:center">Réponse</h3>\n<div class="generation">' + format_references(generated_text) + "</div>"
|
94 |
fiches_html = '<h2 style="text-align:center">Sources</h3>\n' + fiches_html
|