Update app.py
Browse files
app.py
CHANGED
@@ -552,11 +552,11 @@ def generate(message, temperature):
|
|
552 |
If an error occurred, the boolean flag will be True and the content will be the error message.
|
553 |
"""
|
554 |
stream = client.chat.completions.create(
|
555 |
-
model="
|
556 |
messages=message,
|
557 |
stream=False,
|
558 |
max_tokens=1024,
|
559 |
-
stop=["
|
560 |
temperature=temperature,
|
561 |
#timeout=30,
|
562 |
)
|
|
|
552 |
If an error occurred, the boolean flag will be True and the content will be the error message.
|
553 |
"""
|
554 |
stream = client.chat.completions.create(
|
555 |
+
model="H4/zephyr-7b-beta",
|
556 |
messages=message,
|
557 |
stream=False,
|
558 |
max_tokens=1024,
|
559 |
+
stop=["output\n"],
|
560 |
temperature=temperature,
|
561 |
#timeout=30,
|
562 |
)
|