Update app.py

app.py

@@ -101,12 +101,14 @@ def bot(history, api_kind):
          gr.Warning(f"API {api_kind} is not supported")
          raise ValueError(f"API {api_kind} is not supported")
     try:
-        print(prompt_html)
+        count_tokens = lambda text: len([token.strip() for token in text.split() if token.strip()])
+        print(prompt_html,'token count is',count_tokens)
         history[-1][1] = ""
         for character in generate_fn(prompt, history[:-1]):
             history[-1][1] = character
             yield history, prompt_html
     except Exception as e:  # Catch any exception
+        print('An unexpected error occurred during generation:', str(e))
         yield f"An unexpected error occurred during generation: {str(e)}"
 
 with gr.Blocks() as demo: