qmaruf commited on
Commit
0c6afa2
·
1 Parent(s): 3c44f19

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. aussie_bot.py +4 -5
aussie_bot.py CHANGED
@@ -6,10 +6,9 @@ import time
6
 
7
  import gradio as gr
8
  import openai
9
- from langchain import LLMChain, OpenAI, PromptTemplate
10
- from langchain.memory import ConversationBufferWindowMemory
11
  from langchain.chat_models import ChatOpenAI
12
-
13
 
14
  openai.api_key = os.getenv('OPENAI_API_KEY')
15
 
@@ -42,10 +41,10 @@ def get_chain() -> LLMChain:
42
  template=template
43
  )
44
 
45
- llm = ChatOpenAI(model_name='gpt-3.5-turbo', temperature=1.0, )
46
 
47
  chatgpt_chain = LLMChain(
48
- llm=llm,
49
  prompt=prompt,
50
  verbose=True,
51
  memory=ConversationBufferWindowMemory(k=100),
 
6
 
7
  import gradio as gr
8
  import openai
9
+ from langchain import LLMChain, PromptTemplate
 
10
  from langchain.chat_models import ChatOpenAI
11
+ from langchain.memory import ConversationBufferWindowMemory
12
 
13
  openai.api_key = os.getenv('OPENAI_API_KEY')
14
 
 
41
  template=template
42
  )
43
 
44
+ chat = ChatOpenAI(model_name='gpt-3.5-turbo', temperature=1.0)
45
 
46
  chatgpt_chain = LLMChain(
47
+ llm=chat,
48
  prompt=prompt,
49
  verbose=True,
50
  memory=ConversationBufferWindowMemory(k=100),