Update main.py

main.py

@@ -12,8 +12,8 @@ from transformers import pipeline, CLIPProcessor, CLIPModel
 from datetime import datetime
 import uvicorn
 # Hugging Face GPT or LLM model for content-based name generation
-from langchain.chat_models import ChatOpenAI
-import os
+from langchain_openai import ChatOpenAI
+from langchain.schema import HumanMessage
 
 app = FastAPI()