pminervini commited on
Commit
cf0ac44
1 Parent(s): 482d0d4
Files changed (1) hide show
  1. app.py +2 -0
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