Update app.py

app.py

@@ -18,17 +18,18 @@ from langchain_openai import ChatOpenAI
 import re
 from PIL import Image
 
+openai_api_key = os.environ.get("openai_api_key")
 # Cached resources
 @st.cache_resource
 def load_models():
     return {
         "yolo": YOLO("best.pt"),
-        "embeddings": OpenAIEmbeddings(model="text-embedding-3-small"),
-        "llm": ChatOpenAI(model="gpt-4-turbo", temperature=0.3)
+        "embeddings": OpenAIEmbeddings(model="text-embedding-3-small",openai_api_key=openai_api_key),
+        "llm": ChatOpenAI(model="gpt-4-turbo", temperature=0.3,openai_api_key=openai_api_key)
     }
 
 models = load_models()
-openai_api_key = os.environ.get("openai_api_key")
+
 
 # Constants
 FIGURE_CLASS_INDEX = 4