LukasHug commited on
Commit
a2aecc6
1 Parent(s): 9c401b6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -86,6 +86,7 @@ def get_conv_log_filename():
86
 
87
  def get_model_list():
88
  models = [
 
89
  'LukasHug/LlavaGuard-7B-hf',
90
  'LukasHug/LlavaGuard-13B-hf',
91
  'LukasHug/LlavaGuard-34B-hf', ][:1]
@@ -498,11 +499,12 @@ Set the environment variable `model` to change the model:
498
 
499
 
500
  models = [
 
501
  'LukasHug/LlavaGuard-7B-hf',
502
  'LukasHug/LlavaGuard-13B-hf',
503
  'LukasHug/LlavaGuard-34B-hf', ]
504
  bits = int(os.getenv("bits", 16))
505
- model = os.getenv("model", models[1])
506
  available_devices = os.getenv("CUDA_VISIBLE_DEVICES", "0")
507
  model_path, model_name = model, model.split("/")[0]
508
  if api_key:
 
86
 
87
  def get_model_list():
88
  models = [
89
+ 'AIML-TUDA/LlavaGuard-7B',
90
  'LukasHug/LlavaGuard-7B-hf',
91
  'LukasHug/LlavaGuard-13B-hf',
92
  'LukasHug/LlavaGuard-34B-hf', ][:1]
 
499
 
500
 
501
  models = [
502
+ 'AIML-TUDA/LlavaGuard-7B',
503
  'LukasHug/LlavaGuard-7B-hf',
504
  'LukasHug/LlavaGuard-13B-hf',
505
  'LukasHug/LlavaGuard-34B-hf', ]
506
  bits = int(os.getenv("bits", 16))
507
+ model = os.getenv("model", models[0])
508
  available_devices = os.getenv("CUDA_VISIBLE_DEVICES", "0")
509
  model_path, model_name = model, model.split("/")[0]
510
  if api_key: