schuler commited on
Commit
d98cec7
·
verified ·
1 Parent(s): 4377d27

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -110,6 +110,8 @@ def respond(
110
  """
111
  For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
112
  """
 
 
113
  demo = gr.ChatInterface(
114
  respond,
115
  additional_inputs=[
@@ -124,6 +126,7 @@ demo = gr.ChatInterface(
124
  label="Top-p (nucleus sampling)",
125
  ),
126
  ],
 
127
  )
128
 
129
 
 
110
  """
111
  For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
112
  """
113
+ status_text = f"*This chat uses the {REPO_NAME} model with {model.get_memory_footprint() / 1e6:.2f} MB memory footprint.*"
114
+
115
  demo = gr.ChatInterface(
116
  respond,
117
  additional_inputs=[
 
126
  label="Top-p (nucleus sampling)",
127
  ),
128
  ],
129
+ description=status_text
130
  )
131
 
132