pminervini commited on
Commit
5b449d8
1 Parent(s): a447ca5
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -63,7 +63,7 @@ def rag_pipeline(prompt, index="pubmed"):
63
  response = generator(messages, max_new_tokens=256, return_full_text=False)
64
 
65
  # Return the generated text and the documents
66
- return response[0]['generated_text'], "\n\n".join(docs)
67
 
68
  # Create the Gradio interface
69
  iface = gr.Interface(fn=rag_pipeline,
 
63
  response = generator(messages, max_new_tokens=256, return_full_text=False)
64
 
65
  # Return the generated text and the documents
66
+ return response[0]['generated_text'], joined_docs
67
 
68
  # Create the Gradio interface
69
  iface = gr.Interface(fn=rag_pipeline,