Spaces:
Build error
Build error
Upload folder using huggingface_hub
Browse files- 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,
|
| 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 |
-
|
| 46 |
|
| 47 |
chatgpt_chain = LLMChain(
|
| 48 |
-
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),
|