OlivierDehaene commited on
Commit
6728f78
1 Parent(s): df2ff86

add tgi info

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -157,7 +157,7 @@ def radio_on_change(
157
  )
158
 
159
 
160
- title = """<h1 align="center">🔥Large Language Model API 🚀Streaming🚀</h1>"""
161
  description = """Language models can be conditioned to act like dialogue agents through a conversational prompt that typically takes the form:
162
 
163
  ```
@@ -171,6 +171,10 @@ Assistant: <utterance>
171
  In this app, you can explore the outputs of multiple LLMs when prompted in this way.
172
  """
173
 
 
 
 
 
174
  openchat_disclaimer = """
175
  <div align="center">Checkout the official <a href=https://huggingface.co/spaces/togethercomputer/OpenChatKit>OpenChatKit feedback app</a> for the full experience.</div>
176
  """
@@ -180,6 +184,7 @@ with gr.Blocks(
180
  #chatbot {height: 520px; overflow: auto;}"""
181
  ) as demo:
182
  gr.HTML(title)
 
183
  with gr.Column(elem_id="col_container"):
184
  model = gr.Radio(
185
  value="OpenAssistant/oasst-sft-1-pythia-12b",
 
157
  )
158
 
159
 
160
+ title = """<h1 align="center">Large Language Model Chat API</h1>"""
161
  description = """Language models can be conditioned to act like dialogue agents through a conversational prompt that typically takes the form:
162
 
163
  ```
 
171
  In this app, you can explore the outputs of multiple LLMs when prompted in this way.
172
  """
173
 
174
+ text_generation_inference = """
175
+ <div align="center">Powered by: <a href=https://github.com/huggingface/text-generation-inference>Text Generation Inference</a></div>
176
+ """
177
+
178
  openchat_disclaimer = """
179
  <div align="center">Checkout the official <a href=https://huggingface.co/spaces/togethercomputer/OpenChatKit>OpenChatKit feedback app</a> for the full experience.</div>
180
  """
 
184
  #chatbot {height: 520px; overflow: auto;}"""
185
  ) as demo:
186
  gr.HTML(title)
187
+ gr.Markdown(text_generation_inference, visible=True)
188
  with gr.Column(elem_id="col_container"):
189
  model = gr.Radio(
190
  value="OpenAssistant/oasst-sft-1-pythia-12b",