Update setup_code.py

setup_code.py

@@ -188,7 +188,7 @@ class Relevant_Documents_Agent:
           response = get_completion(self.client, prompt)
           count += int(is_Yes(response))
 
-      st.write(count)
+      st.write(str(count))
       # return is_Yes(response)
       return count >= 2