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
|