Update app.py

app.py

@@ -113,10 +113,15 @@ def respond(
 """
 For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
 """
+total_params = sum(p.numel() for p in model.parameters())
+trainable_params = sum(p.numel() for p in model.parameters() if p.requires_grad)
+embed_params = sum(p.numel() for p in model.model.embed_tokens.parameters())*2
+non_embed_params = (trainable_params - embed_params) / 1e6
 cpu_usage = psutil.cpu_percent(interval=1)
 status_text = \
   f"This chat uses the {REPO_NAME} model with {model.get_memory_footprint() / 1e6:.2f} MB memory footprint. " + \
-  f"Current CPU usage is {cpu_usage:.2f}% .'" + \
+  f"Current CPU usage is {cpu_usage:.2f}% . '" + \
+  f"Total number of non embedding trainable parameters: {non_embed_params:.2f} million. " + \
   f"You may ask questions such as 'What is biology?' or 'What is the human body?'"
 
 # """