MatteoScript commited on
Commit
5f3605f
·
verified ·
1 Parent(s): 38f1a79

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -0
main.py CHANGED
@@ -72,7 +72,9 @@ def read_root(request: Request, input_data: InputData):
72
  ]
73
  }}
74
  '''
 
75
  generated_response = generate(input_text, history, temperature, max_new_tokens, top_p, repetition_penalty)
 
76
  return {"response": generated_response}
77
 
78
  def generate(prompt, history, temperature=0.7, max_new_tokens=30000, top_p=0.95, repetition_penalty=1.0):
 
72
  ]
73
  }}
74
  '''
75
+ print(f"{datetime.now()} - Input Text: {input_text}")
76
  generated_response = generate(input_text, history, temperature, max_new_tokens, top_p, repetition_penalty)
77
+ print(f"{datetime.now()} - Response Text: {generated_response}")
78
  return {"response": generated_response}
79
 
80
  def generate(prompt, history, temperature=0.7, max_new_tokens=30000, top_p=0.95, repetition_penalty=1.0):