hadxu commited on
Commit
6f1ca4e
1 Parent(s): c897a05
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -38,12 +38,10 @@ def generate_text(prompt, model = "gpt-3.5-turbo-16k-0613"):
38
  message = clinet.chat.completions.create(
39
  model=model,
40
  messages=[
41
- {"role": "system", "content": "You are a helpful assistant."},
42
- {"role": "assistant", "content": "Here is some initial assistant message."},
43
  {"role": "user", "content": prompt}
44
  ],
45
  max_tokens=max_tokens,
46
- ).choices[0].message['content']
47
  return message
48
 
49
  def generate_answer(question):
 
38
  message = clinet.chat.completions.create(
39
  model=model,
40
  messages=[
 
 
41
  {"role": "user", "content": prompt}
42
  ],
43
  max_tokens=max_tokens,
44
+ ).choices[0].message.content
45
  return message
46
 
47
  def generate_answer(question):