tonic commited on
Commit
c9d06d9
·
1 Parent(s): c5e63bf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -12
app.py CHANGED
@@ -46,12 +46,11 @@ def main():
46
  with gr.Blocks() as demo:
47
  gr.Markdown(title)
48
  with gr.Row():
49
- system_message = gr.Textbox(label="📉System Prompt", placeholder=system_message)
50
- assistant_message = gr.Textbox(label="Assistant Message", placeholder=assistant_message)
51
- tabular_data = gr.Textbox(label="Tabular Data", placeholder=tabular_data)
52
- user_message = gr.Textbox(label="🫡Enter your query here...", placeholder=user_message)
53
 
54
-
55
  with gr.Accordion("Advanced Settings"):
56
  with gr.Row():
57
  max_new_tokens = gr.Slider(label="Max new tokens", value=125, minimum=25, maximum=1250)
@@ -61,14 +60,11 @@ def main():
61
  do_sample = gr.Checkbox(label="Do sample", value=False)
62
 
63
  output_text = gr.Textbox(label="🐯📏TigerAI-StructLM-7B", interactive=True)
64
-
65
- gr.Button("Try🫡📉MetaMath").click(
66
  predict_math_bot,
67
- inputs=[user_message, system_message, assistant_message, tabular_data, max_new_tokens, temperature, top_p, repetition_penalty, do_sample],
68
  outputs=output_text
69
  )
70
-
71
- demo.launch()
72
 
73
- if __name__ == "__main__":
74
- main()
 
46
  with gr.Blocks() as demo:
47
  gr.Markdown(title)
48
  with gr.Row():
49
+ system_message_input = gr.Textbox(label="📉System Prompt", placeholder=system_message) # Renamed variable
50
+ assistant_message_input = gr.Textbox(label="Assistant Message", placeholder=assistant_message) # Renamed variable
51
+ tabular_data_input = gr.Textbox(label="Tabular Data", placeholder=tabular_data) # Renamed variable
52
+ user_message_input = gr.Textbox(label="🫡Enter your query here...", placeholder=user_message) # Renamed variable
53
 
 
54
  with gr.Accordion("Advanced Settings"):
55
  with gr.Row():
56
  max_new_tokens = gr.Slider(label="Max new tokens", value=125, minimum=25, maximum=1250)
 
60
  do_sample = gr.Checkbox(label="Do sample", value=False)
61
 
62
  output_text = gr.Textbox(label="🐯📏TigerAI-StructLM-7B", interactive=True)
63
+
64
+ gr.Button("Try🐯📏TigerAI-StructLM").click(
65
  predict_math_bot,
66
+ inputs=[user_message_input, system_message_input, assistant_message_input, tabular_data_input, max_new_tokens, temperature, top_p, repetition_penalty, do_sample],
67
  outputs=output_text
68
  )
 
 
69
 
70
+ demo.launch()