Carlos Rosas commited on
Commit
e33a35c
·
verified ·
1 Parent(s): 3db8a39

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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