zakerytclarke commited on
Commit
72ae0b1
·
verified ·
1 Parent(s): b8ce407

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -79,7 +79,8 @@ def query_teapot(prompt, context, user_input):
79
  elapsed_time = end_time - start_time
80
  tokens_per_second = total_length / elapsed_time if elapsed_time > 0 else float("inf")
81
 
82
- return f"{output_text} ({tokens_per_second} tokens per second)"
 
83
 
84
 
85
  @log_time
 
79
  elapsed_time = end_time - start_time
80
  tokens_per_second = total_length / elapsed_time if elapsed_time > 0 else float("inf")
81
 
82
+ return f"{output_text}\n({tokens_per_second:.1f} tokens per second)"
83
+
84
 
85
 
86
  @log_time