Spaces:
Sleeping
Sleeping
pminervini
commited on
Commit
•
5b449d8
1
Parent(s):
a447ca5
update
Browse files
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'],
|
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,
|