adi-123 commited on
Commit
db0dcbc
·
1 Parent(s): d477b7f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
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/200 tokens. Finish the story 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
- 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