Update setup_code.py

setup_code.py

@@ -188,9 +188,9 @@ class Relevant_Documents_Agent:
           response = get_completion(self.client, prompt)
           count += int(is_Yes(response))
 
-      st.write(str(count))
+      # st.write(str(count))
       # return is_Yes(response)
-      return count >= 2
+      return count >= 1
 
 class OpenAI_Agent:
   def __init__(self, model="gpt-3.5-turbo", key_filename="open_ai_key.txt"):