Spaces:

wu981526092
/

personality_analysis

Sleeping

Zekun Wu commited on Jul 5

Commit

46a37a0

•

1 Parent(s): 62e10d6

update

Files changed (1) hide show

model.py CHANGED Viewed

@@ -55,7 +55,7 @@ DEFINITION_2 = "\n".join(["### Descriptions ###",
 # Create LLM and Embedding models
 def create_models():
     llm = AzureOpenAI(
-        deployment_name=os.environ.get("AZURE_OPENAI_DEPLOYMENT_ID_RAG"),
         api_key=os.environ.get("OPENAI_API_KEY_RAG"),
         azure_endpoint=os.environ.get("AZURE_OPENAI_ENDPOINT_RAG"),
         api_version="2024-02-01",

 # Create LLM and Embedding models
 def create_models():
     llm = AzureOpenAI(
+        deployment_name="personality_gpt4o",
         api_key=os.environ.get("OPENAI_API_KEY_RAG"),
         azure_endpoint=os.environ.get("AZURE_OPENAI_ENDPOINT_RAG"),
         api_version="2024-02-01",