Tonic commited on
Commit
8503364
1 Parent(s): 36072c8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -52,9 +52,9 @@ def gradio_predict(user_message, system_message, max_new_tokens, temperature, to
52
  Tulu_bot.set_system_message(system_message)
53
  if not do_sample:
54
  max_length = 1269
55
- temperature = 1.2 # Default value
56
- top_p = 0.9 # Default value
57
- repetition_penalty = 0.9 # Default value
58
  response = Tulu_bot.predict(user_message, temperature, max_new_tokens, top_p, repetition_penalty, do_sample)
59
  return response
60
 
@@ -62,8 +62,8 @@ Tulu_bot = TuluChatBot(model, tokenizer)
62
 
63
 
64
  with gr.Blocks(theme = "ParityError/Anime") as demo:
65
- gr.Markdown(title_md) # Display the title
66
- gr.Markdown(description_md) # Display the description
67
  with gr.Row():
68
  system_message = gr.Textbox(label="Optional 🌷Tulu Assistant Message", lines=2)
69
  user_message = gr.Textbox(label="Your Message", lines=3)
 
52
  Tulu_bot.set_system_message(system_message)
53
  if not do_sample:
54
  max_length = 1269
55
+ temperature = 1.2
56
+ top_p = 0.9
57
+ repetition_penalty = 0.9
58
  response = Tulu_bot.predict(user_message, temperature, max_new_tokens, top_p, repetition_penalty, do_sample)
59
  return response
60
 
 
62
 
63
 
64
  with gr.Blocks(theme = "ParityError/Anime") as demo:
65
+ gr.Markdown(title_)
66
+ gr.Markdown(description)
67
  with gr.Row():
68
  system_message = gr.Textbox(label="Optional 🌷Tulu Assistant Message", lines=2)
69
  user_message = gr.Textbox(label="Your Message", lines=3)