Prakh24s commited on
Commit
7bdfb7c
·
verified ·
1 Parent(s): ddead39

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -12,7 +12,6 @@ pp = pprint.PrettyPrinter(indent=4)
12
  def generate_response(messages):
13
  model_name = os.getenv("MODEL_NAME")
14
  response = client.chat.completions.create(model=model_name, messages=messages, temperature=0.5, max_tokens=250)
15
- spinner.stop()
16
  print("Request:")
17
  pp.pprint(messages)
18
  print(f"Completion tokens: {response.usage.completion_tokens}, Prompt tokens: {response.usage.prompt_tokens}, Total tokens: {response.usage.total_tokens}")
 
12
  def generate_response(messages):
13
  model_name = os.getenv("MODEL_NAME")
14
  response = client.chat.completions.create(model=model_name, messages=messages, temperature=0.5, max_tokens=250)
 
15
  print("Request:")
16
  pp.pprint(messages)
17
  print(f"Completion tokens: {response.usage.completion_tokens}, Prompt tokens: {response.usage.prompt_tokens}, Total tokens: {response.usage.total_tokens}")