LukasHug commited on
Commit
98390cc
·
verified ·
1 Parent(s): e0ca52a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -87,9 +87,9 @@ def get_conv_log_filename():
87
 
88
  def get_model_list():
89
  models = [
90
- 'LukasHug/LlavaGuard-7B-hf',
91
- 'LukasHug/LlavaGuard-13B-hf',
92
- 'LukasHug/LlavaGuard-34B-hf', ][:2]
93
  return models
94
 
95
 
@@ -405,10 +405,10 @@ def build_demo(embed_mode, cur_dir=None, concurrency_count=10):
405
  [textbox, upvote_btn, downvote_btn, flag_btn]
406
  )
407
 
408
- model_selector.change(
409
- load_selected_model,
410
- [model_selector],
411
- )
412
 
413
  regenerate_btn.click(
414
  regenerate,
@@ -535,4 +535,4 @@ Set the environment variable `model` to change the model:
535
  print(e)
536
  exit_status = 1
537
  finally:
538
- sys.exit(exit_status)
 
87
 
88
  def get_model_list():
89
  models = [
90
+ 'LukasHug/LlavaGuard-7B-hf',
91
+ 'LukasHug/LlavaGuard-13B-hf',
92
+ 'LukasHug/LlavaGuard-34B-hf', ][:1]
93
  return models
94
 
95
 
 
405
  [textbox, upvote_btn, downvote_btn, flag_btn]
406
  )
407
 
408
+ # model_selector.change(
409
+ # load_selected_model,
410
+ # [model_selector],
411
+ # )
412
 
413
  regenerate_btn.click(
414
  regenerate,
 
535
  print(e)
536
  exit_status = 1
537
  finally:
538
+ sys.exit(exit_status)