LukasHug commited on
Commit
73d561d
·
verified ·
1 Parent(s): 6af9316

Update app.py

Browse files

update model to v1.1

Files changed (1) hide show
  1. app.py +4 -9
app.py CHANGED
@@ -88,9 +88,8 @@ def get_model_list():
88
  models = [
89
  'AIML-TUDA/LlavaGuard-7B',
90
  'AIML-TUDA/LlavaGuard-v1.1-7B-hf',
91
- 'LukasHug/LlavaGuard-7B-hf',
92
- 'LukasHug/LlavaGuard-13B-hf',
93
- 'LukasHug/LlavaGuard-34B-hf', ][1:2]
94
  return models
95
 
96
 
@@ -499,13 +498,9 @@ Set the environment variable `model` to change the model:
499
  api_key = os.getenv("token")
500
 
501
 
502
- models = [
503
- 'AIML-TUDA/LlavaGuard-7B',
504
- 'LukasHug/LlavaGuard-7B-hf',
505
- 'LukasHug/LlavaGuard-13B-hf',
506
- 'LukasHug/LlavaGuard-34B-hf', ]
507
  bits = int(os.getenv("bits", 16))
508
- model = os.getenv("model", models[0])
509
  available_devices = os.getenv("CUDA_VISIBLE_DEVICES", "0")
510
  model_path, model_name = model, model.split("/")[0]
511
  if api_key:
 
88
  models = [
89
  'AIML-TUDA/LlavaGuard-7B',
90
  'AIML-TUDA/LlavaGuard-v1.1-7B-hf',
91
+ 'AIML-TUDA/LlavaGuard-13B',
92
+ 'AIML-TUDA/LlavaGuard-v1.1-13B-hf'][1:2]
 
93
  return models
94
 
95
 
 
498
  api_key = os.getenv("token")
499
 
500
 
501
+ models = get_model_list()
 
 
 
 
502
  bits = int(os.getenv("bits", 16))
503
+ model = os.getenv("model", models[1])
504
  available_devices = os.getenv("CUDA_VISIBLE_DEVICES", "0")
505
  model_path, model_name = model, model.split("/")[0]
506
  if api_key: