jenkim25 commited on
Commit
2f4c75d
Β·
verified Β·
1 Parent(s): f812257

trying to fix system message again

Browse files
Files changed (1) hide show
  1. app.py +11 -5
app.py CHANGED
@@ -139,15 +139,21 @@ with gr.Blocks(css=custom_css) as demo:
139
  gr.Markdown("<h1 style='text-align: center;'>🌟 Fancy AI Chatbot 🌟</h1>")
140
  gr.Markdown("Interact with the AI chatbot using customizable settings below.")
141
 
142
- with gr.Row():
143
- level = gr.Dropdown(choices=["elementary school", "middle school", "high school", "college"])
144
 
145
- system_mess = f"You are a friendly Chatbot. Please give answers at a {level} level"
146
 
147
  with gr.Row():
148
-
 
 
 
 
 
 
 
 
 
149
  use_local_model = gr.Checkbox(label="Use Local Model", value=False)
150
- system_message = gr.Textbox(value=system_mess)
151
 
152
  with gr.Row():
153
  max_tokens = gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens")
 
139
  gr.Markdown("<h1 style='text-align: center;'>🌟 Fancy AI Chatbot 🌟</h1>")
140
  gr.Markdown("Interact with the AI chatbot using customizable settings below.")
141
 
 
 
142
 
 
143
 
144
  with gr.Row():
145
+ system_message = gr.Dropdown(
146
+ choices=["You are a friendly Chatbot that responds at an elementary school level.",
147
+ "You are a friendly middle school Chatbot.", "You are a friendly high school Chatbot.",
148
+ "You are a friendly college Chatbot."],
149
+ value="You are a friendly Chatbot.",
150
+ label="System message",
151
+ interactive=True
152
+ )
153
+
154
+ with gr.Row():
155
  use_local_model = gr.Checkbox(label="Use Local Model", value=False)
156
+
157
 
158
  with gr.Row():
159
  max_tokens = gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens")