Spaces:
Runtime error
Runtime error
Update chain_app.py
Browse files- chain_app.py +5 -3
chain_app.py
CHANGED
@@ -1602,9 +1602,11 @@ async def main(message: cl.Message):
|
|
1602 |
elif chat_profile == 'mistral-nemo-12B':
|
1603 |
client = Client("0x7o/Mistral-Nemo-Instruct", hf_token=hf_token)
|
1604 |
result = client.predict(
|
1605 |
-
|
1606 |
-
|
1607 |
-
|
|
|
|
|
1608 |
)
|
1609 |
await cl.Message(
|
1610 |
content=result[1][0][1]
|
|
|
1602 |
elif chat_profile == 'mistral-nemo-12B':
|
1603 |
client = Client("0x7o/Mistral-Nemo-Instruct", hf_token=hf_token)
|
1604 |
result = client.predict(
|
1605 |
+
message=message.content,
|
1606 |
+
max_new_tokens=512,
|
1607 |
+
temperature=0.7,
|
1608 |
+
top_p=0.95,
|
1609 |
+
api_name="/chat"
|
1610 |
)
|
1611 |
await cl.Message(
|
1612 |
content=result[1][0][1]
|