Update variables.py

variables.py

@@ -95,7 +95,8 @@ def embed_tweets(file,model,query,_prompt):
     ChatOpenAI(temperature=0), 
     chain_type="stuff", 
     vectorstore=docsearch,
-    chain_type_kwargs=chain_type_kwargs
+    chain_type_kwargs=chain_type_kwargs,
+    k=5
     )
 
     result = chain({"query": query})