MatteoScript commited on
Commit
a6e1b57
·
verified ·
1 Parent(s): 1b97fe5

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -0
main.py CHANGED
@@ -54,6 +54,7 @@ def generate_text(request: Request, input_data: InputData):
54
  repetition_penalty = input_data.repetition_penalty
55
  input_text = generate_input_text(input_data)
56
  print(f"{datetime.now()} - Input Text: {input_text}")
 
57
  generated_response = generate(input_text, history, temperature, max_new_tokens, top_p, repetition_penalty)
58
  print(f"{datetime.now()} - Response Text: {generated_response}")
59
  return {"response": generated_response}
 
54
  repetition_penalty = input_data.repetition_penalty
55
  input_text = generate_input_text(input_data)
56
  print(f"{datetime.now()} - Input Text: {input_text}")
57
+ history = []
58
  generated_response = generate(input_text, history, temperature, max_new_tokens, top_p, repetition_penalty)
59
  print(f"{datetime.now()} - Response Text: {generated_response}")
60
  return {"response": generated_response}