Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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)
|