PawinC commited on
Commit
c4b5be4
·
1 Parent(s): 520af46

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -98,7 +98,7 @@ def chat():
98
  print("Got Message" + str(messages))
99
 
100
  if AlpacaLoaded:
101
- modelOutput = llm.create_chat_completion(messages=messages, max_tokens=128)
102
  responseMessage = modelOutput["choices"][0]['message']
103
  print(f"\n\nResponseMessage: {responseMessage}\n\n")
104
  return Response(json.dumps(responseMessage, indent=2), content_type='application/json')
 
98
  print("Got Message" + str(messages))
99
 
100
  if AlpacaLoaded:
101
+ modelOutput = llm.create_chat_completion(messages=messages, max_tokens=512)
102
  responseMessage = modelOutput["choices"][0]['message']
103
  print(f"\n\nResponseMessage: {responseMessage}\n\n")
104
  return Response(json.dumps(responseMessage, indent=2), content_type='application/json')