Spaces:
Sleeping
Sleeping
pminervini
commited on
Commit
•
cf0ac44
1
Parent(s):
482d0d4
update
Browse files
app.py
CHANGED
@@ -73,6 +73,8 @@ def rag_pipeline(prompt, index="pubmed", num_docs=3, model_name="HuggingFaceH4/z
|
|
73 |
|
74 |
# Generate response using the LLM
|
75 |
response = generator(messages, **generation_kwargs)
|
|
|
|
|
76 |
|
77 |
# Return the generated text and the documents
|
78 |
return response[0]['generated_text'], joined_docs
|
|
|
73 |
|
74 |
# Generate response using the LLM
|
75 |
response = generator(messages, **generation_kwargs)
|
76 |
+
|
77 |
+
print('RESPONSE', response)
|
78 |
|
79 |
# Return the generated text and the documents
|
80 |
return response[0]['generated_text'], joined_docs
|