peterkchung commited on
Commit
1331b8a
·
verified ·
1 Parent(s): f1d3421

Update app.py footer with model name

Browse files
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -19,8 +19,9 @@ TODOs:
19
 
20
  """
21
 
 
22
 
23
- client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
24
 
25
  def query_submit(user_message, history):
26
  return "", history + [[user_message, None]]
@@ -100,7 +101,11 @@ with gr.Blocks() as chatUI:
100
  retry = gr.Button("Retry (null)")
101
  clear = gr.ClearButton([queryInput, chatOutput])
102
 
103
- # gr.State()
 
 
 
 
104
 
105
  queryInput.submit(
106
  fn = query_submit,
 
19
 
20
  """
21
 
22
+ selected_model = "mistralai/Mixtral-8x7B-Instruct-v0.1"
23
 
24
+ client = InferenceClient(selected_model)
25
 
26
  def query_submit(user_message, history):
27
  return "", history + [[user_message, None]]
 
101
  retry = gr.Button("Retry (null)")
102
  clear = gr.ClearButton([queryInput, chatOutput])
103
 
104
+ with gr.Row():
105
+
106
+ footer = gr.HTML("<div class='footer'>" + selected_model + "</div>"),
107
+
108
+ # gr.State()
109
 
110
  queryInput.submit(
111
  fn = query_submit,