zting commited on
Commit
4419e3c
1 Parent(s): bbfcc3f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -1,15 +1,17 @@
1
  from langchain.agents import load_tools
2
  from langchain.agents import initialize_agent
3
  from langchain.agents import AgentType
 
4
  from langchain.llms import OpenAI
5
- from langchain.chat_models import ChatOpenAI
6
  from langchain.chains.conversation.memory import ConversationBufferWindowMemory
7
 
8
  import os
9
 
10
- OPENAI_API_KEY = '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'
 
 
 
11
 
12
- llm = ChatOpenAI(openai_api_key=OPENAI_API_KEY, temperature=0, model_name='gpt-3.5-turbo')
13
 
14
  import torch
15
  from transformers import BlipProcessor, BlipForConditionalGeneration
 
1
  from langchain.agents import load_tools
2
  from langchain.agents import initialize_agent
3
  from langchain.agents import AgentType
4
+ from langchain.chat_models import AzureChatOpenAI
5
  from langchain.llms import OpenAI
 
6
  from langchain.chains.conversation.memory import ConversationBufferWindowMemory
7
 
8
  import os
9
 
10
+ OPENAI_API_KEY=os.getenv("OPENAI_API_KEY")
11
+ OPENAI_API_BASE=os.getenv("OPENAI_API_BASE")
12
+ DEP_NAME=os.getenv("deployment_name")
13
+ llm=AzureChatOpenAI(deployment_name=DEP_NAME,openai_api_base=OPENAI_API_BASE,openai_api_key=OPENAI_API_KEY,openai_api_version="2023-03-15-preview",model_name="gpt-3.5-turbo")
14
 
 
15
 
16
  import torch
17
  from transformers import BlipProcessor, BlipForConditionalGeneration