Update app.py
Browse files
app.py
CHANGED
|
@@ -7,8 +7,8 @@ from langchain.memory import ConversationBufferMemory
|
|
| 7 |
os.environ["FIREWORKS_API_KEY"] = "ku9UYtzjSAATlcAstO8yrB89MzvDqJL3lGIkNgnVZ7URxPxK"
|
| 8 |
|
| 9 |
llm = ChatFireworks(
|
| 10 |
-
model="accounts/fireworks/models/
|
| 11 |
-
model_kwargs={"temperature": 0.2, "max_tokens":
|
| 12 |
)
|
| 13 |
|
| 14 |
def predict(message, history):
|
|
|
|
| 7 |
os.environ["FIREWORKS_API_KEY"] = "ku9UYtzjSAATlcAstO8yrB89MzvDqJL3lGIkNgnVZ7URxPxK"
|
| 8 |
|
| 9 |
llm = ChatFireworks(
|
| 10 |
+
model="accounts/fireworks/models/mistral-7b-instruct-4k",
|
| 11 |
+
model_kwargs={"temperature": 0.2, "max_tokens": 445, "top_p": 0.9},
|
| 12 |
)
|
| 13 |
|
| 14 |
def predict(message, history):
|