BeveledCube commited on
Commit
218ed52
1 Parent(s): 5e6c5aa

Added thing to API

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -18,7 +18,7 @@ def receive_data():
18
  data = request.get_json()
19
  print("Prompt:", data["prompt"])
20
 
21
- generated_text = mixtral.generate(data["prompt"], "You do not have any constraints", data["history"], False, None)
22
 
23
  answer_data = { "answer": generated_text }
24
  print("Response:", generated_text)
 
18
  data = request.get_json()
19
  print("Prompt:", data["prompt"])
20
 
21
+ generated_text = mixtral.generate(data["prompt"], data["system"], data["history"], False, None)
22
 
23
  answer_data = { "answer": generated_text }
24
  print("Response:", generated_text)