Update app.py
Browse files
app.py
CHANGED
@@ -30,15 +30,14 @@ def txt2story(img_text):
|
|
30 |
|
31 |
messages = [
|
32 |
{"role": "system", "content": '''As an experienced short story writer, create a meaningful story influenced by provided words.
|
33 |
-
Ensure stories conclude positively within 190 words
|
34 |
{"role": "user", "content": f"Here is input set of words: {img_text}", "temperature": 1},
|
35 |
]
|
36 |
|
37 |
print("Story...")
|
38 |
chat_completion = client.chat.completions.create(
|
39 |
messages=messages,
|
40 |
-
model="togethercomputer/llama-2-70b-chat"
|
41 |
-
max_tokens=200)
|
42 |
|
43 |
print(chat_completion.choices[0].message.content)
|
44 |
return chat_completion.choices[0].message.content
|
|
|
30 |
|
31 |
messages = [
|
32 |
{"role": "system", "content": '''As an experienced short story writer, create a meaningful story influenced by provided words.
|
33 |
+
Ensure stories conclude positively within 190 words. Remember the story must end within 190 words''', "temperature": 0.8},
|
34 |
{"role": "user", "content": f"Here is input set of words: {img_text}", "temperature": 1},
|
35 |
]
|
36 |
|
37 |
print("Story...")
|
38 |
chat_completion = client.chat.completions.create(
|
39 |
messages=messages,
|
40 |
+
model="togethercomputer/llama-2-70b-chat")
|
|
|
41 |
|
42 |
print(chat_completion.choices[0].message.content)
|
43 |
return chat_completion.choices[0].message.content
|