Spaces:

willsh1997
/

widget-RAG

Running on Zero

willsh1997 commited on Aug 25

Commit

e390c83

1 Parent(s): efae5b4

:wrench: tune number of results returned

Files changed (1) hide show

RAG-demo.py CHANGED Viewed

@@ -21,7 +21,7 @@ llama3_pipe = pipeline(
 #google search setup
 def google_search_results(input_question: str):
-    outputGenerator = search(input_question, num_results=3, advanced=True)
     outputs = []
     for result in outputGenerator:
         outputs.append(result.description)

 #google search setup
 def google_search_results(input_question: str):
+    outputGenerator = search(input_question, num_results=5, advanced=True)
     outputs = []
     for result in outputGenerator:
         outputs.append(result.description)