Update app.py
Browse files
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
|
|
|
|
|
|
|
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
|