Tonic commited on
Commit
fe35217
1 Parent(s): 52c03eb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -69,8 +69,15 @@ def gradio_Tulu(user_message, system_message, max_new_tokens, temperature, top_p
69
  response = Tulu_bot.Tulu(user_message, temperature, max_new_tokens, top_p, repetition_penalty, do_sample)
70
  return response
71
 
 
72
  Tulu_bot = TuluChatBot(model, tokenizer)
73
 
 
 
 
 
 
 
74
 
75
  with gr.Blocks(theme = "ParityError/Anime") as demo:
76
  gr.Markdown(title)
 
69
  response = Tulu_bot.Tulu(user_message, temperature, max_new_tokens, top_p, repetition_penalty, do_sample)
70
  return response
71
 
72
+ # Initialize TuluChatBot
73
  Tulu_bot = TuluChatBot(model, tokenizer)
74
 
75
+ # Gradio interface function
76
+ def gradio_Tulu(user_message, system_message, max_new_tokens, temperature, top_p, repetition_penalty, do_sample):
77
+ Tulu_bot.set_system_message(system_message)
78
+ response = Tulu_bot.Tulu(user_message, temperature, max_new_tokens, top_p, repetition_penalty, do_sample)
79
+ return response
80
+
81
 
82
  with gr.Blocks(theme = "ParityError/Anime") as demo:
83
  gr.Markdown(title)