Ferdi commited on
Commit
c1ecd4a
1 Parent(s): 2014dfb

add load button back to interface

Browse files
Files changed (2) hide show
  1. src/app.py +6 -1
  2. 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
- #model_load_btn.click(load_models, [llm], load_success_msg, api_name="load_models")
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