Update generator.py

generator.py

@@ -3,12 +3,7 @@ import time
 import os
 
 # Set your OpenAI API key 
-openai_api_key = os.getenv("OPENAI_API_KEY")
-# Check if the API key is set correctly
-if openai_api_key is None:
-    print("API key is not set!")
-else:
-    print("API key loaded successfully!")
+openai.api_key = os.getenv("OPENAI_API_KEY")
 
 
 def generate_response_from_document(query, retrieved_docs):