Update app.py
Browse files
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 |
|