Update llm_loader.py

llm_loader.py

@@ -1,12 +1,11 @@
-# llm_loader.py
-from langchain.chat_models import ChatOpenAI
-
-def load_model(openai_api_key):
-    return ChatOpenAI(
-        model_name="gpt-4o",
-        openai_api_key=openai_api_key,
-        temperature=0.01,
-        max_tokens=2096,
-        top_p=0.95,
-        top_k=25
+from langchain_community.chat_models import ChatOpenAI
+
+def load_model(openai_api_key):
+    return ChatOpenAI(
+        model_name="gpt-4o",
+        openai_api_key=openai_api_key,
+        temperature=0.01,
+        max_tokens=2096,
+        top_p=0.95,
+        top_k=25
     )