add load button back to interface
Browse files- src/app.py +6 -1
- src/utils.py +1 -6
src/app.py
CHANGED
@@ -32,6 +32,11 @@ with gr.Blocks(gr.themes.Soft(primary_hue=gr.themes.colors.slate, secondary_hue=
|
|
32 |
model_load_btn = gr.Button('Load model', variant='primary',scale=2)
|
33 |
load_success_msg = gr.Textbox(show_label=False,lines=1, placeholder="Model loading ...")
|
34 |
"""
|
|
|
|
|
|
|
|
|
|
|
35 |
chatbot = gr.Chatbot([], elem_id="chatbot",
|
36 |
label='Chatbox', height=725, )
|
37 |
|
@@ -45,7 +50,7 @@ with gr.Blocks(gr.themes.Soft(primary_hue=gr.themes.colors.slate, secondary_hue=
|
|
45 |
with gr.Column(scale=1):
|
46 |
clear_btn = gr.Button('Clear',variant='stop',size = 'sm')
|
47 |
|
48 |
-
|
49 |
|
50 |
txt.submit(add_text, [chatbot, txt], [chatbot, txt]).then(
|
51 |
bot, [chatbot,instruction,temperature,max_new_tokens,k_context], chatbot)
|
|
|
32 |
model_load_btn = gr.Button('Load model', variant='primary',scale=2)
|
33 |
load_success_msg = gr.Textbox(show_label=False,lines=1, placeholder="Model loading ...")
|
34 |
"""
|
35 |
+
with gr.Row(equal_height=True):
|
36 |
+
llm="gpt-3.5-turbo"
|
37 |
+
model_load_btn = gr.Button('Load model', variant='primary',scale=2)
|
38 |
+
load_success_msg = gr.Textbox(show_label=False,lines=1, placeholder="Model loading ...")
|
39 |
+
|
40 |
chatbot = gr.Chatbot([], elem_id="chatbot",
|
41 |
label='Chatbox', height=725, )
|
42 |
|
|
|
50 |
with gr.Column(scale=1):
|
51 |
clear_btn = gr.Button('Clear',variant='stop',size = 'sm')
|
52 |
|
53 |
+
model_load_btn.click(load_models, [llm], load_success_msg, api_name="load_models")
|
54 |
|
55 |
txt.submit(add_text, [chatbot, txt], [chatbot, txt]).then(
|
56 |
bot, [chatbot,instruction,temperature,max_new_tokens,k_context], chatbot)
|
src/utils.py
CHANGED
@@ -55,9 +55,4 @@ def bot(history,
|
|
55 |
def clear_cuda_cache():
|
56 |
|
57 |
gc.collect()
|
58 |
-
return None
|
59 |
-
|
60 |
-
def load_prompts_list_from_json(json_filepath):
|
61 |
-
with open(json_filepath, 'r') as file:
|
62 |
-
data = json.load(file)
|
63 |
-
return list(data.keys())
|
|
|
55 |
def clear_cuda_cache():
|
56 |
|
57 |
gc.collect()
|
58 |
+
return None
|
|
|
|
|
|
|
|
|
|